bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update DQN

Browse Source
This commit is contained in:
JohnJim0816
2020-11-28 10:16:25 +08:00
parent abfe6ea62b
commit 59a09144f1
6 changed files with 41 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
codes/dqn/utils.py
View File

@@ -5,7 +5,7 @@ Author: John
Email: johnjim0816@gmail.com
Date: 2020-10-15 21:28:00
LastEditor: John
LastEditTime: 2020-10-15 21:50:30
LastEditTime: 2020-10-30 16:56:55
Discription:
Environment:
'''
@@ -14,8 +14,17 @@ import numpy as np
def save_results(rewards,moving_average_rewards,ep_steps,tag='train',result_path='./result'):
'''保存reward等结果
'''
if not os.path.exists(result_path): # 检测是否存在文件夹
os.mkdir(result_path)
np.save(result_path+'rewards_'+tag+'.npy', rewards)
np.save(result_path+'moving_average_rewards_'+tag+'.npy', moving_average_rewards)
np.save(result_path+'steps_'+tag+'.npy',ep_steps )
np.save(result_path+'steps_'+tag+'.npy',ep_steps )
print('results saved!')
def save_model(agent,model_path='./saved_model'):
if not os.path.exists(model_path): # 检测是否存在文件夹
os.mkdir(model_path)
agent.save_model(model_path+'checkpoint.pth')
print('model saved！')