Python load_checkpoint Examples

Example #1

File: players.py Project: Veyron2121/rl_games

 def restore(self, fn):
     checkpoint = torch_ext.load_checkpoint(fn)
     self.model.sac_network.actor.load_state_dict(checkpoint['actor'])
     self.model.sac_network.critic.load_state_dict(checkpoint['critic'])
     self.model.sac_network.critic_target.load_state_dict(
         checkpoint['critic_target'])
     if self.normalize_input:
         self.running_mean_std.load_state_dict(
             checkpoint['running_mean_std'])

Example #2

 def restore(self, fn):
     checkpoint = torch_ext.load_checkpoint(fn)
     self.set_full_state_weights(checkpoint)

Example #3

File: players.py Project: Veyron2121/rl_games

 def restore(self, fn):
     checkpoint = torch_ext.load_checkpoint(fn)
     self.model.load_state_dict(checkpoint['model'])
     if self.normalize_input:
         self.running_mean_std.load_state_dict(
             checkpoint['running_mean_std'])