Fix the learning_rate=None issue, see facenet issue davidsandberg#884.

xunings · Apr 17, 2019 · 4661780 · 4661780
1 parent 60f3086
commit 4661780
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/facenet.py b/src/facenet.py
@@ -307,7 +307,9 @@ def get_learning_rate_from_file(filename, epoch):
                 if e <= epoch:
                     learning_rate = lr
                 else:
-                    return learning_rate
+                    #return learning_rate
+                    break
+        return learning_rate
 
 class ImageClass():
     "Stores the paths to images for a given class"