fix training issue

Author: bourdakos1

train.ipynb

@@ -6,11 +6,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%%bash\n",
-    "\n",
-    "pip install -U \"tensorflow~=2.0\" \"tensorflow-hub[make_image_classifier]~=0.6\"\n",
-    "\n",
-    "make_image_classifier \\\n",
+    "!pip install \"tensorflow~=2.0\" \"tensorflow-hub[make_image_classifier]~=0.6\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!make_image_classifier \\\n",
     "  --image_dir data \\\n",
     "  --tfhub_module https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4 \\\n",
     "  --image_size 224 \\\n",
@@ -25,9 +30,16 @@
     "  --l1_regularizer=0.0 \\\n",
     "  --l2_regularizer=0.0001 \\\n",
     "  --label_smoothing=0.1 \\\n",
-    "  --validation_split=0.2\n",
-    "\n",
-    "mv labels.txt model"
+    "  --validation_split=0.2"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!mv labels.txt model"
    ]
   }
  ],
@@ -52,4 +64,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}