hishamcse commited on
Commit
eb9a964
·
verified ·
1 Parent(s): 375223b

Upload config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +5 -5
config.yaml CHANGED
@@ -8,8 +8,8 @@ settings:
8
  frame_shape: !!python/tuple [128, 128, 1]
9
  continue_game: 0.0
10
  action_space: "discrete"
11
- characters: "Ken"
12
- difficulty: 6
13
  outfits: 2
14
 
15
  wrappers_settings:
@@ -31,7 +31,7 @@ policy_kwargs:
31
 
32
  ppo_settings:
33
  gamma: 0.94
34
- model_checkpoint: "100000" # 0: No checkpoint, 100000: Load checkpoint (if previously trained for 100000 steps)
35
  learning_rate: [2.5e-4, 2.5e-6] # To start
36
  clip_range: [0.15, 0.025] # To start
37
  #learning_rate: [5.0e-5, 2.5e-6] # Fine Tuning
@@ -39,5 +39,5 @@ ppo_settings:
39
  batch_size: 512 #8 #nminibatches gave different batch size depending on the number of environments: batch_size = (n_steps * n_envs) // nminibatches
40
  n_epochs: 4
41
  n_steps: 512
42
- autosave_freq: 10000
43
- time_steps: 500000
 
8
  frame_shape: !!python/tuple [128, 128, 1]
9
  continue_game: 0.0
10
  action_space: "discrete"
11
+ characters: !!python/tuple ["Ken", "Ryu"]
12
+ difficulty: 7
13
  outfits: 2
14
 
15
  wrappers_settings:
 
31
 
32
  ppo_settings:
33
  gamma: 0.94
34
+ model_checkpoint: "600000" # 0: No checkpoint, else: Load checkpoint (if previously trained)
35
  learning_rate: [2.5e-4, 2.5e-6] # To start
36
  clip_range: [0.15, 0.025] # To start
37
  #learning_rate: [5.0e-5, 2.5e-6] # Fine Tuning
 
39
  batch_size: 512 #8 #nminibatches gave different batch size depending on the number of environments: batch_size = (n_steps * n_envs) // nminibatches
40
  n_epochs: 4
41
  n_steps: 512
42
+ autosave_freq: 50000
43
+ time_steps: 750000