AI4Finance-Foundation/ElegantRL

A ERROR about test

Opened this issue · 1 comments

guest-oo commented

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

stellawang196 commented

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.