fixed hpc save, load. cleaned apu

2026-09-09 11:32:07 +08:00 · 2019-07-26 22:09:35 -04:00
parent 64de447545
commit 348223a702
1 changed files with 1 additions and 1 deletions
@@ -79,7 +79,7 @@ class TrainerIO(object):

        # add the state_dict from the model
        model = self.__get_model()
-        checkpoint['state_dict'] = model.get_state_dict
+        checkpoint['state_dict'] = model.state_dict()

        # give the model a chance to add a few things
        model.on_save_checkpoint(checkpoint)