File size: 810 Bytes
a444d1c |
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 35 36 37 38 39 40 41 42 |
!!python/object/apply:collections.OrderedDict
- - - batch_size
- 512
- - clip_range
- 0.1
- - ent_coef
- 0.0010159833764878474
- - gae_lambda
- 0.98
- - gamma
- 0.995
- - learning_rate
- 0.0003904770450788824
- - max_grad_norm
- 0.9
- - n_envs
- 1
- - n_epochs
- 20
- - n_steps
- 2048
- - n_timesteps
- 1000000.0
- - normalize
- gamma: 0.995
norm_obs: false
norm_reward: true
- - policy
- MlpPolicy
- - policy_kwargs
- activation_fn: !!python/name:torch.nn.modules.activation.ReLU ''
features_extractor_class: !!python/name:imitation.policies.base.NormalizeFeaturesExtractor ''
net_arch:
- pi:
- 64
- 64
vf:
- 64
- 64
- - vf_coef
- 0.20315938606555833
|