File size: 721 Bytes
62e03a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
general_cfg:
  algo_name: GAIL
  batch_size: 2048
  device: cuda
  discount: 0.99
  env_name: CartPole-v1
  eval_eps: 10
  eval_per_episode: 5
  hidden_dim: 32
  imitation_batch_size: 128
  imitation_epochs: 5
  imitation_replay_size: 4
  load_checkpoint: true
  load_path: Train_CartPole-v1_GAIL_20221207-160945
  lr: 0.001
  max_steps: 200
  mode: test
  new_step_api: true
  ppo_epochs: 5
  r1_reg_coeff: 1
  render: false
  save_fig: true
  seed: 1
  show_fig: false
  test_eps: 10
  train_eps: 400
  wrapper: null
algo_cfg:
  actor_hidden_dim: 256
  actor_lr: 0.0003
  batch_size: 2048
  critic_hidden_dim: 256
  critic_lr: 0.0003
  entropy_coef: 0.01
  eps_clip: 0.2
  gamma: 0.99
  k_epochs: 4
  update_freq: 2048