Closed jc-bao closed 1 year ago
Using agent.act.load_state_dict(loaded_agent['actor'].state_dict()) to load policy rather than agent.act = loaded_agent['actor'] where the policy gradient is disabled.
agent.act.load_state_dict(loaded_agent['actor'].state_dict())
agent.act = loaded_agent['actor']