diff --git a/expman b/expman index 032a9ca..7653d4a 160000 --- a/expman +++ b/expman @@ -1 +1 @@ -Subproject commit 032a9ca2c5d116b90e971d7307cfcd414a5893d4 +Subproject commit 7653d4a11940796d39e9ee9ac3f21512629276b2 diff --git a/train.py b/train.py index 5047e1a..d1f5946 100644 --- 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_' \