Skip to content

Commit

Permalink
added simple config
Browse files Browse the repository at this point in the history
  • Loading branch information
DenSumy committed Dec 6, 2023
1 parent 005ec17 commit 6fce694
Showing 1 changed file with 82 additions and 0 deletions.
82 changes: 82 additions & 0 deletions rl_games/configs/mario/mario_v1.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
params:
seed: 322
algo:
name: a2c_discrete

model:
name: discrete_a2c

network:
name: actor_critic
separate: False
space:
discrete:

cnn:
#permute_input: False
type: conv2d
activation: elu
initializer:
name: default
#name: glorot_normal_initializer
#gain: 1.4142
regularizer:
name: None
convs:
- filters: 32
kernel_size: 8
strides: 4
padding: 0
- filters: 64
kernel_size: 4
strides: 2
padding: 0
- filters: 64
kernel_size: 3
strides: 1
padding: 0

mlp:
units: [512]
activation: elu
initializer:
name: orthogonal_initializer
gain: 1.41421356237

config:
name: mario_ray
env_name: 'SuperMarioBros-v1'
score_to_win: 20.0
normalize_value: True
normalize_input: True
reward_shaper:
min_val: -1
max_val: 1
normalize_advantage: True
gamma: 0.99
tau: 0.95
grad_norm: 1.0
entropy_coef: 0.01
truncate_grads: True
learning_rate: 3e-4
lr_schedule: adaptive
kl_threshold: 0.01
grad_norm: 1.0
entropy_coef: 0.01
truncate_grads: True
e_clip: 0.2
clip_value: False
num_actors: 64
horizon_length: 128
minibatch_size: 2048
mini_epochs: 4
critic_coef: 2
max_epochs: 500
use_diagnostics: False


player:
render: True
games_num: 100
n_game_life: 1
deterministic: True

0 comments on commit 6fce694

Please sign in to comment.