yann-j's picture
Initial commit
7b03cd8
raw
history blame contribute delete
585 Bytes
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 32
- - buffer_size
- 100000
- - env_wrapper
- - stable_baselines3.common.atari_wrappers.AtariWrapper
- - exploration_final_eps
- 0.05
- - exploration_fraction
- 0.1
- - exploration_initial_eps
- 1.0
- - frame_stack
- 4
- - gradient_steps
- 1
- - learning_rate
- 0.0002
- - learning_starts
- 100
- - n_timesteps
- 1000000
- - optimize_memory_usage
- false
- - policy
- CnnPolicy
- - target_update_interval
- 10000
- - train_freq
- 4