Fixed AE weights loading issue
Fixed cuda error by loading AE weights explicitly on CPU. Otherwise there was an issue when loading saved client/server weights that included the autoencoder.
Fixed cuda error by loading AE weights explicitly on CPU. Otherwise there was an issue when loading saved client/server weights that included the autoencoder.