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

update

Browse Source
This commit is contained in:
johnjim0816
2021-09-15 10:32:52 +08:00
parent 95f3f4dd57
commit 5085040330
74 changed files with 431 additions and 433 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
codes/DoubleDQN/task0_train.py
View File

@@ -5,7 +5,7 @@
@Email: johnjim0816@gmail.com
@Date: 2020-06-12 00:48:57
@LastEditor: John
LastEditTime: 2021-05-04 22:26:59
LastEditTime: 2021-09-10 15:26:05
@Discription:
@Environment: python 3.7.7
'''
@@ -106,7 +106,7 @@ def eval(cfg,env,agent):
if __name__ == "__main__":
cfg = DoubleDQNConfig()
# train
# 训练
env,agent = env_agent_config(cfg,seed=1)
rewards, ma_rewards = train(cfg, env, agent)
make_dir(cfg.result_path, cfg.model_path)
@@ -115,7 +115,7 @@ if __name__ == "__main__":
plot_rewards(rewards, ma_rewards, tag="train",
algo=cfg.algo, path=cfg.result_path)
# eval
# 测试
env,agent = env_agent_config(cfg,seed=10)
agent.load(path=cfg.model_path)
rewards,ma_rewards = eval(cfg,env,agent)