Commit 9a9546af authored by Vít Musil's avatar Vít Musil
Browse files

bugfix: checkpointing

parent 495be430

trainer.py

+1 −1
Original line number Diff line number Diff line
@@ -97,7 +97,7 @@ class StrategyTrainer(object): 
            val_train=self.val_train,

            loss=self.loss.item(),

            strategy=self.strategy,

            state_dict=deepcopy(self.game.state_dict()),

            state_dict=deepcopy(self.game.strategy.state_dict()),

            extras={**self.extras, **self.eval_extras},

            **self.times)