updated expman

fabiocarrara · Dec 22, 2020 · 4f54154 · 4f54154
1 parent 46c4972
commit 4f54154
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/expman b/expman
diff --git a/train.py b/train.py
@@ -50,8 +50,8 @@ def main(args):
     model.compile(optimizer='sgd', loss='binary_crossentropy', metrics={'out_tags': 'binary_accuracy'})
 
     log = exp.path_to('log.csv')
-    best_ckpt_path = exp.ckpt('best.h5')
-    last_ckpt_path = exp.ckpt('last.h5')
+    best_ckpt_path = exp.path_to('best_weights.h5')
+    last_ckpt_path = exp.path_to('last_weights.h5')
 
     # today = datetime.datetime.now().strftime('%Y-%m-%d')
     best_model_path = 'meye-segmentation_' \