fix path

J13253902839 · Aug 17, 2020 · beec5e6 · beec5e6
1 parent 048884e
commit beec5e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -24,7 +24,7 @@ def main(cfg):
     checkpoint_dir.mkdir(exist_ok=True)
     callbacks = [
         LearningRateScheduler(schedule=scheduler),
-        ModelCheckpoint(str(checkpoint_dir) + "weights.{epoch:02d}-{val_loss:.2f}.hdf5",
+        ModelCheckpoint(str(checkpoint_dir) + "/weights.{epoch:02d}-{val_loss:.2f}.hdf5",
                         monitor="val_loss",
                         verbose=1,
                         save_best_only=True,