From 2be89f96031f4c9c4d69765108b3bb5db8e6b1bd Mon Sep 17 00:00:00 2001 From: Archit-Kohli <118905854+Archit-Kohli@users.noreply.github.com> Date: Wed, 4 Oct 2023 19:00:43 +0530 Subject: [PATCH] Minor Comments Update in pix2pix.ipynb Updated comments in upsampling part of the generator to include dimensions due to concatenation because of skip connections for easier understanding --- site/en/tutorials/generative/pix2pix.ipynb | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/site/en/tutorials/generative/pix2pix.ipynb b/site/en/tutorials/generative/pix2pix.ipynb index 5912fab9be3..b338500b522 100644 --- a/site/en/tutorials/generative/pix2pix.ipynb +++ b/site/en/tutorials/generative/pix2pix.ipynb @@ -637,12 +637,12 @@ "\n", " up_stack = [\n", " upsample(512, 4, apply_dropout=True), # (batch_size, 2, 2, 1024)\n", - " upsample(512, 4, apply_dropout=True), # (batch_size, 4, 4, 1024)\n", - " upsample(512, 4, apply_dropout=True), # (batch_size, 8, 8, 1024)\n", - " upsample(512, 4), # (batch_size, 16, 16, 1024)\n", - " upsample(256, 4), # (batch_size, 32, 32, 512)\n", - " upsample(128, 4), # (batch_size, 64, 64, 256)\n", - " upsample(64, 4), # (batch_size, 128, 128, 128)\n", + " upsample(512, 4, apply_dropout=True), # (batch_size, 4, 4, 512) + (batch_size, 4, 4, 512) = (batch_size, 4, 4, 1024)\n", + " upsample(512, 4, apply_dropout=True), # (batch_size, 8, 8, 512) + (batch_size, 8, 8, 512) = (batch_size, 8, 8, 1024)\n", + " upsample(512, 4), # (batch_size, 16, 16, 512) + (batch_size, 16, 16, 512) = (batch_size, 16, 16, 1024)\n", + " upsample(256, 4), # (batch_size, 32, 32, 256) + (batch_size, 32, 32, 256) = (batch_size, 32, 32, 512)\n", + " upsample(128, 4), # (batch_size, 64, 64, 128) + (batch_size, 64, 64, 128) = (batch_size, 64, 64, 256)\n", + " upsample(64, 4), # (batch_size, 128, 128, 64) + (batch_size, 128, 128, 64) = (batch_size, 128, 128, 128)\n", " ]\n", "\n", " initializer = tf.random_normal_initializer(0., 0.02)\n",