LorMolf commited on
Commit
4476660
·
verified ·
1 Parent(s): 761e852

Upload latest retained checkpoint envstep_300000.pth.tar

Browse files
Files changed (1) hide show
  1. 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': 10,
175
  'replay_ratio': 0.25,
176
- 'batch_size': 128,
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': 20,
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/simplified5_constant_lr_fixed_20260519/round-01/three_six_nine/attempt-01_260519_093513/ckpt/envstep_90000.pth.tar',
223
  'reanalyze_ratio': 0.0,
224
- 'eval_freq': 100001,
225
- 'replay_buffer_size': 50000,
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/three_six_nine_dual_eval_continuation_20260519/round-01/three_six_nine/attempt-01_260519_131901',
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
  }