File size: 732 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
40
41
42
43
44
general_cfg:
  algo_name: DQN
  device: cpu
  env_name: MountainCar-v0
  eval_eps: 10
  eval_per_episode: 5
  load_checkpoint: false
  load_path: tasks
  max_steps: 200
  mode: train
  new_step_api: true
  render: false
  save_fig: true
  seed: 3
  show_fig: false
  test_eps: 20
  train_eps: 200
  wrapper: null
algo_cfg:
  batch_size: 64
  buffer_size: 40000
  epsilon_decay: 800
  epsilon_end: 0.01
  epsilon_start: 0.99
  gamma: 0.99
  lr: 0.01
  target_update: 4
  value_layers:
  - activation: relu
    layer_dim:
    - n_states
    - 256
    layer_type: linear
  - activation: relu
    layer_dim:
    - 256
    - 256
    layer_type: linear
  - activation: none
    layer_dim:
    - 256
    - n_actions
    layer_type: linear