Update segmentation.ipynb

Golbstein · Aug 10, 2019 · b16282a · b16282a
1 parent 1d37c21
commit b16282a
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/segmentation.ipynb b/segmentation.ipynb
@@ -30,7 +30,6 @@
     "bs = 2\n",
     "\n",
     "better_model = False\n",
-    "load_pretrained_weights = True\n",
     "\n",
     "losses = sparse_crossentropy_ignoring_last_label\n",
     "metrics = {'pred_mask' : [Jaccard, sparse_accuracy_ignoring_last_label]}\n",
@@ -122,14 +121,13 @@
     "\n",
     "# fine-tune model (train only last conv layers)\n",
     "flag = 0\n",
-    "if load_pretrained_weights:\n"
-    "    for k, l in enumerate(model.layers):\n",
-    "        l.trainable = False\n",
-    "        if l.name == 'concat_projection':\n",
-    "            flag = 1\n",
-    "        if flag:\n",
-    "            l.trainable = True\n",
-    "            \n",
+    " for k, l in enumerate(model.layers):\n",
+    "    l.trainable = False\n",
+    "    if l.name == 'concat_projection':\n",
+    "        flag = 1\n",
+    "    if flag:\n",
+    "        l.trainable = True\n",
+    "        \n",
     "def build_callbacks(tf_board = False):\n",
     "    tensorboard = TensorBoard(log_dir='./logs/'+SegClass.net, histogram_freq=0,\n",
     "                        write_graph=False, write_images = False)\n",