Update vanilla_gan.ipynb

Author: StoneT2000

SP21/GAN/vanilla_gan.ipynb

@@ -12,6 +12,7 @@
     "import torch.nn.functional as F\n",
     "import torch\n",
     "import torchvision\n",
+    "from torchvision import transforms\n",
     "import matplotlib.pyplot as plt\n",
     "from tqdm import notebook"
    ]
@@ -103,7 +104,8 @@
     "transform = transforms.Compose([\n",
     "    transforms.ToTensor(),\n",
     "    transforms.Normalize(mean=(0.5), std=(0.5))])\n",
-    "train_dataset = torchvision.datasets.MNIST(root=\"./data\", train = True, download=True, transform=transform)der(train_dataset, batch_size=batch_size, shuffle=True)"
+    "train_dataset = torchvision.datasets.MNIST(root=\"./data\", train = True, download=True, transform=transform)\n",
+    "train_dataloader = torch.utils.data.DataLoader(train_dataset, batch_size=batch_size, shuffle=True)"
    ]
   },
   {
@@ -204,9 +206,8 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
-   "language": "python",
-   "name": "python3"
+   "name": "python379jvsc74a57bd0c3e162dbea2a7bde87c7844ddb2ca1843aba08611dc435aaff358cdc82c48661",
+   "display_name": "Python 3.7.9 64-bit"
   },
   "language_info": {
    "codemirror_mode": {
@@ -218,9 +219,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.9"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}