File size: 585 Bytes
b4a035e |
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 |
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 512
- - buffer_size
- 1000000
- - env_wrapper
- sb3_contrib.common.wrappers.TimeFeatureWrapper
- - gamma
- 0.98
- - learning_rate
- 0.001
- - n_timesteps
- 1000000.0
- - policy
- MultiInputPolicy
- - policy_kwargs
- dict(net_arch=[512, 512, 512], n_critics=2)
- - replay_buffer_class
- HerReplayBuffer
- - replay_buffer_kwargs
- dict( online_sampling=True, goal_selection_strategy='future', n_sampled_goal=4,
max_episode_length=100 )
- - tau
- 0.005
|