Change state saving to every epoch

2020-03-07 18:03:19 -05:00 · 2020-03-07 18:03:19 -05:00 · 93d973b5c8
commit 93d973b5c8
parent ccb323e6ee
1 changed files with 1 additions and 5 deletions
--- a/map2map/train.py
+++ b/map2map/train.py
@ -261,11 +261,7 @@ def gpu_worker(local_rank, node, args):
            state_file = 'state_{}.pth'
            torch.save(state, ckpt_file)
            del state
-
-            if good:
            shutil.copyfile(ckpt_file, state_file.format(epoch + 1))
-                #if os.path.isfile(state_file.format(epoch)):
-                #    os.remove(state_file.format(epoch))

    dist.destroy_process_group()