dqn-SpaceInvaders / config.yml
Subarashi's picture
Initial commit
94b9cb4 verified
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 32
- - buffer_size
- 100000
- - env_wrapper
- - stable_baselines3.common.atari_wrappers.AtariWrapper
- - exploration_final_eps
- 0.01
- - exploration_fraction
- 0.05
- - frame_stack
- 4
- - gradient_steps
- 2
- - learning_rate
- 0.0003
- - learning_starts
- 50000
- - n_timesteps
- 1000000.0
- - optimize_memory_usage
- false
- - policy
- CnnPolicy
- - target_update_interval
- 2000
- - train_freq
- 4