From 11fbc34ab90c72ff3305f4dcfae851e3e6d992d6 Mon Sep 17 00:00:00 2001
From: Marco Leonhardt <marco.leonhardt@rwth-aachen.de>
Date: Fri, 25 Oct 2019 14:35:57 +0200
Subject: [PATCH] Fixed wrong transfer flag

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 293dacb7..f683d459 100644
--- 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: