From bfae0216e13327179fa1f04f28a7d9a7fc5dc7a6 Mon Sep 17 00:00:00 2001 From: Aditya Trivedi Date: Sat, 10 Oct 2020 00:11:33 +0530 Subject: [PATCH 1/2] Updated the auto-encoder example. Fixes #18712 --- .../convolutional_autoencoder.ipynb | 61 +++++++++---------- 1 file changed, 28 insertions(+), 33 deletions(-) diff --git a/example/autoencoder/convolutional_autoencoder.ipynb b/example/autoencoder/convolutional_autoencoder.ipynb index a49eba0fcc10..a18ee558cdac 100644 --- a/example/autoencoder/convolutional_autoencoder.ipynb +++ b/example/autoencoder/convolutional_autoencoder.ipynb @@ -108,41 +108,36 @@ "metadata": {}, "outputs": [], "source": [ - "net = gluon.nn.HybridSequential(prefix='autoencoder_')\n", - "with net.name_scope():\n", - " # Encoder 1x28x28 -> 32x1x1\n", - " encoder = gluon.nn.HybridSequential(prefix='encoder_')\n", - " with encoder.name_scope():\n", - " encoder.add(\n", - " gluon.nn.Conv2D(channels=4, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=8, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=16, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=32, kernel_size=3, padding=0, strides=(2,2),activation='relu'),\n", - " gluon.nn.BatchNorm()\n", - " )\n", - " decoder = gluon.nn.HybridSequential(prefix='decoder_')\n", - " # Decoder 32x1x1 -> 1x28x28\n", - " with decoder.name_scope():\n", - " decoder.add(\n", - " gluon.nn.Conv2D(channels=32, kernel_size=3, padding=2, activation='relu'),\n", - " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=16, kernel_size=3, padding=1, activation='relu'),\n", - " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=8, kernel_size=3, padding=2, activation='relu'),\n", - " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", - " gluon.nn.BatchNorm(),\n", - " gluon.nn.Conv2D(channels=4, kernel_size=3, padding=1, activation='relu'),\n", - " gluon.nn.Conv2D(channels=1, kernel_size=3, padding=1, activation='sigmoid')\n", - " )\n", - " net.add(\n", + "net = gluon.nn.HybridSequential()\n", + "encoder = gluon.nn.HybridSequential()\n", + "encoder.add(\n", + " gluon.nn.Conv2D(channels=4, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=8, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=16, kernel_size=3, padding=1, strides=(2,2), activation='relu'),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=32, kernel_size=3, padding=0, strides=(2,2),activation='relu'),\n", + " gluon.nn.BatchNorm()\n", + ")\n", + "decoder = gluon.nn.HybridSequential()\n", + "decoder.add(\n", + " gluon.nn.Conv2D(channels=32, kernel_size=3, padding=2, activation='relu'),\n", + " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=16, kernel_size=3, padding=1, activation='relu'),\n", + " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=8, kernel_size=3, padding=2, activation='relu'),\n", + " gluon.nn.HybridLambda(lambda F, x: F.UpSampling(x, scale=2, sample_type='nearest')),\n", + " gluon.nn.BatchNorm(),\n", + " gluon.nn.Conv2D(channels=4, kernel_size=3, padding=1, activation='relu'),\n", + " gluon.nn.Conv2D(channels=1, kernel_size=3, padding=1, activation='sigmoid')\n", + ")\n", + "net.add(\n", " encoder,\n", " decoder\n", - " )" + ")" ] }, { From a8690fbd390ac1807f4d149e21f38d11243383d7 Mon Sep 17 00:00:00 2001 From: Aditya Trivedi Date: Sat, 10 Oct 2020 00:38:46 +0530 Subject: [PATCH 2/2] add Aditya Trivedi to contributors --- CONTRIBUTORS.md | 1 + 1 file changed, 1 insertion(+) diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md index c9fe77ce4a09..54fb1075fe85 100644 --- a/CONTRIBUTORS.md +++ b/CONTRIBUTORS.md @@ -137,6 +137,7 @@ List of Contributors -------------------- * [Top-100 Contributors](https://github.com/apache/incubator-mxnet/graphs/contributors) - To contributors: please add your name to the list when you submit a patch to the project:) +* [Aditya Trivedi](https://github.com/iadi7ya) * [Feng Wang](https://github.com/happynear) - Feng makes MXNet compatible with Windows Visual Studio. * [Jack Deng](https://github.com/jdeng)