-
Notifications
You must be signed in to change notification settings - Fork 354
Open
Description
There are policy.actor and policy.critic in base_runner.py, but I couldn't find them in the algorithms folder of MPE. How can I solve this problem?
def restore(self):
for agent_id in range(self.num_agents):
policy_actor_state_dict = torch.load(str(self.model_dir) + '/actor_agent' + str(agent_id) + '.pt')
self.policy[agent_id].get_actions.load_state_dict(policy_actor_state_dict)
policy_critic_state_dict = torch.load(str(self.model_dir) + '/critic_agent' + str(agent_id) + '.pt')
self.policy[agent_id].get_values.load_state_dict(policy_critic_state_dict)
if self.trainer[agent_id]._use_valuenorm:
policy_vnrom_state_dict = torch.load(str(self.model_dir) + '/vnrom_agent' + str(agent_id) + '.pt')
self.trainer[agent_id].value_normalizer.load_state_dict(policy_vnrom_state_dict)
Metadata
Metadata
Assignees
Labels
No labels