Merge pull request zzh8829#83 from MarBra110/master

Fixed wrong transfer flag
bhardwajRahul · Nov 14, 2019 · 6cf777b · 6cf777b
2 parents 4cd9a9e + 11fbc34
commit 6cf777b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -83,7 +83,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: