Fixed VQVAE solution comments

bipiniiith · Mar 6, 2020 · 84ec318 · 84ec318
1 parent afc78fe
commit 84ec318
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeworks/hw3/hw3_notebook.ipynb b/homeworks/hw3/hw3_notebook.ipynb
@@ -920,7 +920,7 @@
         "    - a (# of epochs + 1,) numpy array of VQ-VAE train losses evaluated once at initialization and after each epoch\n",
         "    - a (# of training iterations,) numpy array of PixelCNN prior train losess evaluated every minibatch\n",
         "    - a (# of epochs + 1,) numpy array of PixelCNN prior train losses evaluated once at initialization and after each epoch\n",
-        "    - a (100, 32, 32, 3) numpy array of 100 samples (an equal number from each class) with values in {0, ... 255}\n",
+        "    - a (100, 32, 32, 3) numpy array of 100 samples with values in {0, ... 255}\n",
         "    - a (100, 32, 32, 3) numpy array of 50 real image / reconstruction pairs\n",
         "      FROM THE TEST SET with values in [0, 255]\n",
         "    \"\"\"\n",
@@ -1084,7 +1084,7 @@
         "    - a (# of epochs + 1,) numpy array of VQ-VAE train losses evaluated once at initialization and after each epoch\n",
         "    - a (# of training iterations,) numpy array of PixelCNN prior train losess evaluated every minibatch\n",
         "    - a (# of epochs + 1,) numpy array of PixelCNN prior train losses evaluated once at initialization and after each epoch\n",
-        "    - a (100, 32, 32, 3) numpy array of 100 samples (an equal number from each class) with values in {0, ... 255}\n",
+        "    - a (100, 32, 32, 3) numpy array of 100 samples with values in {0, ... 255}\n",
         "    - a (100, 32, 32, 3) numpy array of 50 real image / reconstruction pairs\n",
         "      FROM THE TEST SET with values in [0, 255]\n",
         "    \"\"\"\n",