dqn_spaceInvaders-2 / config.yml
saeedHedayatian's picture
Initial commit
66fc8d1
raw
history blame
549 Bytes
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 64
- - buffer_size
- 150000
- - env_wrapper
- - stable_baselines3.common.atari_wrappers.AtariWrapper
- - exploration_final_eps
- 0.02
- - exploration_fraction
- 0.1
- - frame_stack
- 4
- - gradient_steps
- 1
- - learning_rate
- 0.0003
- - learning_starts
- 100000
- - n_timesteps
- 2000000.0
- - optimize_memory_usage
- false
- - policy
- CnnPolicy
- - target_update_interval
- 1000
- - train_freq
- 4