Merge pull request #394 from whisperingkahuna/patch-1

Update cifar10_augmentation.ipynb
udacity · Mar 2, 2021 · 9a094dd · 9a094dd
2 parents 72751f4 + ba1c8a2
commit 9a094dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/projects/practice_projects/cnn/cifar10-augmentation/cifar10_augmentation.ipynb b/projects/practice_projects/cnn/cifar10-augmentation/cifar10_augmentation.ipynb
@@ -256,7 +256,7 @@
     "\n",
     "# visualize augmented images\n",
     "fig = plt.figure(figsize=(20,2))\n",
-    "for x_batch in datagen.flow(x_train_subset, batch_size=12):\n",
+    "for x_batch in datagen_train.flow(x_train_subset, batch_size=12):\n",
     "    for i in range(0, 12):\n",
     "        ax = fig.add_subplot(1, 12, i+1)\n",
     "        ax.imshow(x_batch[i])\n",