Fixed wrong transfer flag

zcdliuwei · Oct 25, 2019 · 11fbc34 · 11fbc34
1 parent 4a11d93
commit 11fbc34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -79,7 +79,7 @@ def main(_argv):
             # freeze darknet
             darknet = model.get_layer('yolo_darknet')
             freeze_all(darknet)
-        elif FLAGS.mode == 'frozen':
+        elif FLAGS.transfer == 'frozen':
             # freeze everything
             freeze_all(model)
         else: