hishamcse commited on
Commit
26ae7c3
·
verified ·
1 Parent(s): b26639a

Upload config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +43 -0
config.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ folders:
2
+ parent_dir: "./results/"
3
+ model_name: "sr6_128x4_das_nc"
4
+
5
+ settings:
6
+ game_id: "sfiii3n"
7
+ step_ratio: 6
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:
16
+ normalize_reward: true
17
+ no_attack_buttons_combinations: true
18
+ stack_frames: 4
19
+ dilation: 1
20
+ add_last_action: true
21
+ stack_actions: 12
22
+ scale: true
23
+ exclude_image_scaling: true
24
+ role_relative: true
25
+ flatten: true
26
+ filter_keys: ["action", "own_health", "opp_health", "own_side", "opp_side", "opp_character", "stage", "timer"]
27
+
28
+ policy_kwargs:
29
+ #net_arch: [{ pi: [64, 64], vf: [32, 32] }]
30
+ net_arch: [64, 64]
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
38
+ #clip_range: [0.075, 0.025] # 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: 10000
43
+ time_steps: 10000