File size: 788 Bytes
cca14b2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
!!python/object/apply:collections.OrderedDict
- - - batch_size
    - 16
  - - clip_range
    - 0.3
  - - ent_coef
    - 3.1441389214159857e-06
  - - gae_lambda
    - 0.8
  - - gamma
    - 0.995
  - - learning_rate
    - 0.00017959211641976886
  - - max_grad_norm
    - 0.9
  - - n_epochs
    - 10
  - - 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.Tanh ''
      features_extractor_class: !!python/name:imitation.policies.base.NormalizeFeaturesExtractor ''
      net_arch:
      - pi:
        - 64
        - 64
        vf:
        - 64
        - 64
  - - vf_coef
    - 0.4351450387648799