log

mnist_classifier.ipynb

@@ -11,11 +11,23 @@
     "import torch.nn.functional as F\n",
     "from datasets import load_dataset\n",
     "import fastcore.all as fc\n",
-    "import matplotlib.pyplot as plt\n",
-    "import matplotlib as mpl\n",
     "import torchvision.transforms.functional as TF\n",
     "from torch.utils.data import default_collate, DataLoader\n",
-    "import torch.optim as optim\n"
+    "import torch.optim as optim"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "tags": [
+     "exclude"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "import matplotlib.pyplot as plt\n",
+    "import matplotlib as mpl"
    ]
   },
   {

mnist_classifier.py

@@ -6,8 +6,6 @@ from torch import nn
 import torch.nn.functional as F
 from datasets import load_dataset
 import fastcore.all as fc
-import matplotlib.pyplot as plt
-import matplotlib as mpl
 import torchvision.transforms.functional as TF
 from torch.utils.data import default_collate, DataLoader
 import torch.optim as optim