zjowowen commited on
Commit
e73bd6b
1 Parent(s): 74ddfea

Upload policy_config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. policy_config.py +24 -0
policy_config.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exp_config = {
2
+ 'type': 'ppo',
3
+ 'on_policy': True,
4
+ 'cuda': True,
5
+ 'action_space': 'discrete',
6
+ 'discount_factor': 0.99,
7
+ 'gae_lambda': 0.95,
8
+ 'epoch_per_collect': 10,
9
+ 'batch_size': 64,
10
+ 'learning_rate': 0.0003,
11
+ 'weight_decay': 0,
12
+ 'value_weight': 0.5,
13
+ 'entropy_weight': 0.01,
14
+ 'clip_ratio': 0.2,
15
+ 'adv_norm': True,
16
+ 'value_norm': 'popart',
17
+ 'ppo_param_init': True,
18
+ 'grad_norm': 0.5,
19
+ 'n_sample': 400,
20
+ 'unroll_len': 1,
21
+ 'deterministic_eval': True,
22
+ 'model': {},
23
+ 'cfg_type': 'PPOFPolicyDict'
24
+ }