mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-19 17:17:04 -05:00
25 lines
611 B
YAML
25 lines
611 B
YAML
cartpole-ppo:
|
|
env: CartPole-v1
|
|
run: PPO
|
|
stop:
|
|
episode_reward_mean: 475
|
|
time_total_s: 300
|
|
checkpoint_config:
|
|
checkpoint_frequency: 2
|
|
checkpoint_at_end: true
|
|
config:
|
|
# Works for both torch and tf.
|
|
framework: torch
|
|
gamma: 0.99
|
|
lr: 0.0003
|
|
num_workers: 1
|
|
observation_filter: MeanStdFilter
|
|
num_sgd_iter: 6
|
|
vf_loss_coeff: 0.01
|
|
model:
|
|
fcnet_hiddens: [32]
|
|
fcnet_activation: linear
|
|
vf_share_layers: true
|
|
enable_connectors: true
|
|
render_env: true
|