| 348 | print("====================================") |
| 349 | |
| 350 | def load(self): |
| 351 | self.policy_net.load_state_dict(torch.load('./pytorch_models/policy_net.pth')) |
| 352 | self.value_net.load_state_dict(torch.load ('./pytorch_models/value_net.pth')) |
| 353 | self.Q_net1.load_state_dict(torch.load ('./pytorch_models/Q_net1.pth')) |
| 354 | self.Q_net2.load_state_dict(torch.load ('./pytorch_models/Q_net2.pth')) |
| 355 | print("====================================") |
| 356 | print("model has been loaded...") |
| 357 | print("====================================") |