Update vae_keras_cnn.py

CSilentLearner · Jun 10, 2019 · d6fc517 · d6fc517
1 parent 080d90d
commit d6fc517
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/vae_keras_cnn.py b/vae_keras_cnn.py
@@ -20,12 +20,6 @@
 from keras.datasets import mnist
 
 
-batch_size = 100
-latent_dim = 2 # 隐变量取2维只是为了方便后面画图
-intermediate_dim = 256
-epochs = 50
-
-
 # 加载MNIST数据集
 (x_train, y_train_), (x_test, y_test_) = mnist.load_data()
 
@@ -38,12 +32,13 @@
 
 # 网络参数
 input_shape = (image_size, image_size, 1)
-batch_size = 128
+batch_size = 100
 kernel_size = 3
 filters = 16
-latent_dim = 2
+latent_dim = 2 # 隐变量取2维只是为了方便后面画图
 epochs = 30
 
+
 x_in = Input(shape=input_shape)
 x = x_in