Skip to content
This repository has been archived by the owner on Jun 26, 2021. It is now read-only.

Commit

Permalink
Merge pull request #147 from NKPmedia/fixed_save_name
Browse files Browse the repository at this point in the history
WIP: Fixed name of saved net that the trainer tries to load
  • Loading branch information
justusschock committed Jun 17, 2019
2 parents 21de652 + f5b55b2 commit 6feb661
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion delira/training/base_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -776,7 +776,7 @@ def _search_for_prev_state(path, extensions=None):
for x in files])

latest_state_path = [x for x in files
if x.startswith("checkpoint_%d"
if x.startswith("checkpoint_epoch_%d"
% latest_epoch)][0]

return latest_state_path, latest_epoch
Expand Down

0 comments on commit 6feb661

Please sign in to comment.