Upload latest retained checkpoint envstep_300000.pth.tar
Browse files- metadata/total_config.py +7 -7
metadata/total_config.py
CHANGED
|
@@ -171,9 +171,9 @@ exp_config = {
|
|
| 171 |
'use_augmentation': False,
|
| 172 |
'augmentation': ['shift', 'intensity'],
|
| 173 |
'ignore_done': False,
|
| 174 |
-
'update_per_collect':
|
| 175 |
'replay_ratio': 0.25,
|
| 176 |
-
'batch_size':
|
| 177 |
'optim_type': 'Adam',
|
| 178 |
'learning_rate': 0.003,
|
| 179 |
'target_update_freq': 100,
|
|
@@ -183,7 +183,7 @@ exp_config = {
|
|
| 183 |
'grad_clip_value': 0.5,
|
| 184 |
'n_episode': 4,
|
| 185 |
'num_segments': 8,
|
| 186 |
-
'num_simulations':
|
| 187 |
'discount_factor': 1,
|
| 188 |
'td_steps': 81,
|
| 189 |
'num_unroll_steps': 5,
|
|
@@ -219,10 +219,10 @@ exp_config = {
|
|
| 219 |
'type': 'efficientzero',
|
| 220 |
'import_names': ['lzero.policy.efficientzero'],
|
| 221 |
'model_path':
|
| 222 |
-
'/workspace/combinatorial_reasoning_post_training/models/
|
| 223 |
'reanalyze_ratio': 0.0,
|
| 224 |
-
'eval_freq':
|
| 225 |
-
'replay_buffer_size':
|
| 226 |
'best_ckpt_strategy': 'raw',
|
| 227 |
'best_ckpt_ema_alpha': 0.3,
|
| 228 |
'best_ckpt_min_episodes': 20,
|
|
@@ -241,6 +241,6 @@ exp_config = {
|
|
| 241 |
'device': 'cuda'
|
| 242 |
},
|
| 243 |
'exp_name':
|
| 244 |
-
'models/
|
| 245 |
'seed': 0
|
| 246 |
}
|
|
|
|
| 171 |
'use_augmentation': False,
|
| 172 |
'augmentation': ['shift', 'intensity'],
|
| 173 |
'ignore_done': False,
|
| 174 |
+
'update_per_collect': 50,
|
| 175 |
'replay_ratio': 0.25,
|
| 176 |
+
'batch_size': 256,
|
| 177 |
'optim_type': 'Adam',
|
| 178 |
'learning_rate': 0.003,
|
| 179 |
'target_update_freq': 100,
|
|
|
|
| 183 |
'grad_clip_value': 0.5,
|
| 184 |
'n_episode': 4,
|
| 185 |
'num_segments': 8,
|
| 186 |
+
'num_simulations': 50,
|
| 187 |
'discount_factor': 1,
|
| 188 |
'td_steps': 81,
|
| 189 |
'num_unroll_steps': 5,
|
|
|
|
| 219 |
'type': 'efficientzero',
|
| 220 |
'import_names': ['lzero.policy.efficientzero'],
|
| 221 |
'model_path':
|
| 222 |
+
'/workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-01_260520_133229/ckpt/envstep_10000.pth.tar',
|
| 223 |
'reanalyze_ratio': 0.0,
|
| 224 |
+
'eval_freq': 300001,
|
| 225 |
+
'replay_buffer_size': 100000,
|
| 226 |
'best_ckpt_strategy': 'raw',
|
| 227 |
'best_ckpt_ema_alpha': 0.3,
|
| 228 |
'best_ckpt_min_episodes': 20,
|
|
|
|
| 241 |
'device': 'cuda'
|
| 242 |
},
|
| 243 |
'exp_name':
|
| 244 |
+
'/workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-03_260520_145623',
|
| 245 |
'seed': 0
|
| 246 |
}
|