File size: 521 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
general_cfg:
  algo_name: DQN_CNN
  device: cuda
  env_name: Mario
  eval_eps: 10
  eval_per_episode: 5
  load_checkpoint: false
  load_path: Train_Mario_DQN_CNN_20221205-205203
  max_steps: 1000
  mode: train
  new_step_api: true
  render: false
  save_fig: true
  seed: 1
  show_fig: false
  test_eps: 10
  train_eps: 200
  wrapper: envs.wrappers.MarioWrappers
algo_cfg:
  batch_size: 64
  buffer_size: 100000
  epsilon_decay: 500
  epsilon_end: 0.01
  epsilon_start: 0.95
  gamma: 0.99
  lr: 0.0001
  target_update: 4