Updated Transfer learning labs

NthabisengMoela · Aug 19, 2020 · 3fd83b2 · 3fd83b2
1 parent 5f846c8
commit 3fd83b2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/05c - Transfer Learning (PyTorch).ipynb b/05c - Transfer Learning (PyTorch).ipynb
@@ -448,8 +448,8 @@
   ],
   "metadata": {
     "kernelspec": {
-      "name": "python_defaultSpec_1597787964136",
-      "display_name": "Python 3.7.6 64-bit ('base': conda)",
+      "name": "python3",
+      "display_name": "Python 3",
       "language": "python"
     },
     "language_info": {

diff --git a/05c - Transfer Learning (Tensorflow).ipynb b/05c - Transfer Learning (Tensorflow).ipynb
@@ -58,7 +58,7 @@
         "data_folder = 'data/shapes'\n",
         "# Our source images are 128x128, but the base model we're going to use was trained with 224x224 images\n",
         "pretrained_size = (224,224)\n",
-        "batch_size = 30\n",
+        "batch_size = 15\n",
         "\n",
         "print(\"Getting Data...\")\n",
         "datagen = ImageDataGenerator(rescale=1./255, # normalize pixel values\n",
@@ -148,7 +148,7 @@
       },
       "cell_type": "code",
       "source": [
-        "# Train the model over 5 epochs using 30-image batches and using the validation holdout dataset for validation\n",
+        "# Train the model over 5 epochs\n",
         "num_epochs = 5\n",
         "history = model.fit(\n",
         "    train_generator,\n",
@@ -312,8 +312,8 @@
   ],
   "metadata": {
     "kernelspec": {
-      "name": "python_defaultSpec_1597791820390",
-      "display_name": "Python 3.7.6 64-bit ('base': conda)",
+      "name": "python3",
+      "display_name": "Python 3",
       "language": "python"
     },
     "language_info": {