added clean slurm save load test

2026-09-09 11:32:07 +08:00 · 2019-07-26 22:26:55 -04:00
parent b5419fcd8b
commit ffa7a0dbab
1 changed files with 1 additions and 0 deletions
@@ -181,6 +181,7 @@ class TrainerIO(object):
        # call model hook
        model.on_hpc_load(checkpoint)

+        self.model = model

    def max_ckpt_in_folder(self, path):
        files = os.listdir(path)