updated notebook

.gitignore

@@ -130,3 +130,4 @@ dmypy.json
 
 # Datasets
 datasets/
+**.h5

notebooks/enhance_me_train.ipynb

@@ -34,6 +34,7 @@
    },
    "outputs": [],
    "source": [
+    "import os\n",
     "import sys\n",
     "\n",
     "sys.path.append(\"./enhance-me\")\n",
@@ -142,6 +143,15 @@
     "history = mirnet.train(epochs=epochs)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "mirnet.load_weights(os.path.join(mirnet.experiment_name, \"weights.h5\"))"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,