A ERROR about test

[guest-oo]

当使用demo_DQN_Dueling_Double_DQN 训练结束的的pt文件无法作为测试时的权重文件 ，是否需要将保存pt文件
由torch.save(actor, actor_path)
更改为torch.save(actor.state_dict(), actor_path)

[stellawang196]

Hi, have you tried to load the pt file by using torch.load(file_path)? In addition, there is method save_or_load_agent() in the AgentBase class, which you may utilize.