fix calls to save_networks, and load_networks, have strides actually …

Author: Kyle Stewart-Frantz

code/model_training.ipynb

@@ -89,7 +89,7 @@
     "    return output_layer\n",
     "\n",
     "def conv2d_batchnorm(input_layer, filters, kernel_size=3, strides=1):\n",
-    "    output_layer = layers.Conv2D(filters=filters, kernel_size=kernel_size, strides=1, \n",
+    "    output_layer = layers.Conv2D(filters=filters, kernel_size=kernel_size, strides=strides, \n",
     "                      padding='same', activation='relu')(input_layer)\n",
     "    \n",
     "    output_layer = layers.BatchNormalization()(output_layer) \n",
@@ -325,7 +325,7 @@
    "source": [
     "# Save your trained model weights\n",
     "weight_file_name = 'model_weights'\n",
-    "model_tools.save_network(weight_file_name)"
+    "model_tools.save_network(model, weight_file_name)"
    ]
   },
   {
@@ -352,7 +352,8 @@
    "source": [
     "# If you need to load a model which you previously trained you can uncomment the codeline that calls the function below.\n",
     "\n",
-    "# model_tools.load_network(weight_file_name)"
+    "# weight_file_name = 'model_weights'\n",
+    "# restored_model = model_tools.load_network(weight_file_name)"
    ]
   },
   {
@@ -529,7 +530,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [default]",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -543,7 +544,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.2"
+   "version": "3.6.2"
   },
   "widgets": {
    "state": {},