Upload policy_config.py with huggingface_hub
Browse files- policy_config.py +7 -5
policy_config.py
CHANGED
@@ -11,10 +11,11 @@ exp_config = {
|
|
11 |
'cfg_type': 'BaseEnvManagerDict'
|
12 |
},
|
13 |
'stop_value': -250,
|
|
|
|
|
14 |
'collector_env_num': 10,
|
15 |
'evaluator_env_num': 8,
|
16 |
-
'act_scale': True
|
17 |
-
'n_evaluator_episode': 8
|
18 |
},
|
19 |
'policy': {
|
20 |
'model': {
|
@@ -67,9 +68,10 @@ exp_config = {
|
|
67 |
'render_freq': -1,
|
68 |
'mode': 'train_iter'
|
69 |
},
|
|
|
70 |
'cfg_type': 'InteractionSerialEvaluatorDict',
|
71 |
-
'
|
72 |
-
'
|
73 |
}
|
74 |
},
|
75 |
'other': {
|
@@ -78,7 +80,7 @@ exp_config = {
|
|
78 |
}
|
79 |
},
|
80 |
'on_policy': False,
|
81 |
-
'cuda':
|
82 |
'multi_gpu': False,
|
83 |
'bp_update_sync': True,
|
84 |
'traj_len_inf': False,
|
|
|
11 |
'cfg_type': 'BaseEnvManagerDict'
|
12 |
},
|
13 |
'stop_value': -250,
|
14 |
+
'n_evaluator_episode': 8,
|
15 |
+
'env_id': 'Pendulum-v1',
|
16 |
'collector_env_num': 10,
|
17 |
'evaluator_env_num': 8,
|
18 |
+
'act_scale': True
|
|
|
19 |
},
|
20 |
'policy': {
|
21 |
'model': {
|
|
|
68 |
'render_freq': -1,
|
69 |
'mode': 'train_iter'
|
70 |
},
|
71 |
+
'figure_path': None,
|
72 |
'cfg_type': 'InteractionSerialEvaluatorDict',
|
73 |
+
'stop_value': -250,
|
74 |
+
'n_episode': 8
|
75 |
}
|
76 |
},
|
77 |
'other': {
|
|
|
80 |
}
|
81 |
},
|
82 |
'on_policy': False,
|
83 |
+
'cuda': True,
|
84 |
'multi_gpu': False,
|
85 |
'bp_update_sync': True,
|
86 |
'traj_len_inf': False,
|