Upload config.yaml
Browse files- 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:
|
13 |
outfits: 2
|
14 |
|
15 |
wrappers_settings:
|
@@ -31,7 +31,7 @@ policy_kwargs:
|
|
31 |
|
32 |
ppo_settings:
|
33 |
gamma: 0.94
|
34 |
-
model_checkpoint: "
|
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:
|
43 |
-
time_steps:
|
|
|
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
|