File size: 592 Bytes
3c4ef74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 128
- - buffer_size
- 100000
- - env_wrapper
- - stable_baselines3.common.atari_wrappers.AtariWrapper
- - exploration_final_eps
- 0.01
- - exploration_fraction
- 0.1
- - frame_stack
- 4
- - gamma
- 0.99
- - gradient_steps
- 1
- - learning_rate
- 0.025
- - learning_starts
- 10000
- - n_envs
- 8
- - n_timesteps
- 5000000.0
- - optimize_memory_usage
- false
- - policy
- CnnPolicy
- - target_update_interval
- 1000
- - train_freq
- 4
|