Skip to content

Commit 8cc464c

Browse files
committed
docs: add conv example for gan using conv2dtranspose
1 parent 754ae5b commit 8cc464c

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/models-usages/generation/gan-image-generation/gan-mnist-convolutional.ipynb

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"\n",
1818
"from neuralnetlib.preprocessing import one_hot_encode\n",
1919
"from neuralnetlib.models import Sequential, GAN\n",
20-
"from neuralnetlib.layers import Input, Dense, Conv2D, Reshape, Flatten, UpSampling2D"
20+
"from neuralnetlib.layers import Input, Dense, Conv2D, Reshape, Flatten, Conv2DTranspose"
2121
]
2222
},
2323
{
@@ -67,10 +67,8 @@
6767
"generator.add(Input(noise_dim))\n",
6868
"generator.add(Dense(7 * 7 * 128))\n",
6969
"generator.add(Reshape((7, 7, 128)))\n",
70-
"generator.add(UpSampling2D(size=(2, 2))) # 14x14\n",
71-
"generator.add(Conv2D(64, kernel_size=3, padding='same', activation='relu'))\n",
72-
"generator.add(UpSampling2D(size=(2, 2))) # 28x28\n",
73-
"generator.add(Conv2D(32, kernel_size=3, padding='same', activation='relu'))\n",
70+
"generator.add(Conv2DTranspose(64, kernel_size=3, strides=2, padding='same', activation='relu'))\n",
71+
"generator.add(Conv2DTranspose(32, kernel_size=3, strides=2, padding='same', activation='relu'))\n",
7472
"generator.add(Conv2D(1, kernel_size=3, padding='same', activation='sigmoid'))"
7573
]
7674
},

0 commit comments

Comments
 (0)