更新算法模版

2022-11-06 12:15:36 +08:00
parent 466a17707f
commit dc78698262
256 changed files with 17282 additions and 10229 deletions
--- a/projects/codes/DQN/config/Acrobot-v1_DQN_Test.yaml
+++ b/projects/codes/DQN/config/Acrobot-v1_DQN_Test.yaml
@@ -0,0 +1,22 @@
+general_cfg:
+  algo_name: DQN
+  device: cuda
+  env_name: Acrobot-v1
+  mode: test
+  load_checkpoint: true
+  load_path: Train_Acrobot-v1_DQN_20221026-094645
+  max_steps: 100000
+  save_fig: true
+  seed: 1
+  show_fig: false
+  test_eps: 10
+  train_eps: 100
+algo_cfg:
+  batch_size: 128
+  buffer_size: 200000
+  epsilon_decay: 500
+  epsilon_end: 0.01
+  epsilon_start: 0.95
+  gamma: 0.95
+  lr: 0.002 
+  target_update: 4