Upload latest retained checkpoint envstep_300000.pth.tar
Browse files- metadata/metadata.yaml +215 -0
metadata/metadata.yaml
ADDED
|
@@ -0,0 +1,215 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
metadata_version: 2
|
| 2 |
+
saved_at: '2026-05-21T05:09:21.229267+00:00'
|
| 3 |
+
entrypoint: src.train_game
|
| 4 |
+
game: three_six_nine
|
| 5 |
+
algo: efficientzero
|
| 6 |
+
seed: 0
|
| 7 |
+
max_env_step: 300000
|
| 8 |
+
exp_name: /workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-03_260520_145623
|
| 9 |
+
checkpoint_dir: /workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-03_260520_145623/ckpt
|
| 10 |
+
config_path: /workspace/combinatorial_reasoning_post_training/data/supported_training/runs/simplified5_two_machine_continuation_20260520/shard-0/configs/three_six_nine.yaml
|
| 11 |
+
cli_args:
|
| 12 |
+
game: null
|
| 13 |
+
config: /workspace/combinatorial_reasoning_post_training/data/supported_training/runs/simplified5_two_machine_continuation_20260520/shard-0/configs/three_six_nine.yaml
|
| 14 |
+
list_games: false
|
| 15 |
+
algo: null
|
| 16 |
+
num_simulations: null
|
| 17 |
+
batch_size: null
|
| 18 |
+
update_per_collect: null
|
| 19 |
+
lr: null
|
| 20 |
+
seed: null
|
| 21 |
+
max_env_step: null
|
| 22 |
+
num_res_blocks: null
|
| 23 |
+
num_channels: null
|
| 24 |
+
replay_buffer_size: null
|
| 25 |
+
num_unroll_steps: null
|
| 26 |
+
reanalyze_ratio: null
|
| 27 |
+
battle_mode: null
|
| 28 |
+
no_wandb: false
|
| 29 |
+
cpu: false
|
| 30 |
+
torch_num_threads: 4
|
| 31 |
+
torch_num_interop_threads: 1
|
| 32 |
+
no_checkpoints: false
|
| 33 |
+
ckpt_dir: /workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-03
|
| 34 |
+
model_path: null
|
| 35 |
+
ckpt_every_env_steps: null
|
| 36 |
+
max_checkpoints_to_keep: null
|
| 37 |
+
best_ckpt_strategy: null
|
| 38 |
+
best_ckpt_ema_alpha: null
|
| 39 |
+
best_ckpt_min_episodes: null
|
| 40 |
+
eval_every_train_iters: null
|
| 41 |
+
eval_every_env_steps: null
|
| 42 |
+
eval_every_env_step_ratio: null
|
| 43 |
+
eval_opponent_type: null
|
| 44 |
+
eval_opponent_checkpoint_path: null
|
| 45 |
+
eval_opponent_checkpoint_selector: null
|
| 46 |
+
eval_opponent_update_policy: null
|
| 47 |
+
eval_opponent_num_simulations: null
|
| 48 |
+
eval_opponent_episodes: null
|
| 49 |
+
eval_opponent_env_num: null
|
| 50 |
+
eval_opponent_promotion_threshold: null
|
| 51 |
+
eval_opponent_fallback_to_env_bot: false
|
| 52 |
+
wandb_project: crpt-simplified5-corrected-quality
|
| 53 |
+
wandb_run_name: simplified5_two_machine_continuation_20260520__shard-0__three_six_nine__a03
|
| 54 |
+
wandb_group: simplified5_two_machine_continuation_20260520
|
| 55 |
+
wandb_run_id: crpt-3367d2521da9e2789a06
|
| 56 |
+
wandb_dir: /workspace/combinatorial_reasoning_post_training/data/supported_training/runs/simplified5_two_machine_continuation_20260520/shard-0/wandb
|
| 57 |
+
source_config:
|
| 58 |
+
game: three_six_nine
|
| 59 |
+
algo: efficientzero
|
| 60 |
+
wandb_project: crpt
|
| 61 |
+
wandb_enabled: true
|
| 62 |
+
checkpoints:
|
| 63 |
+
save: true
|
| 64 |
+
every_env_steps: 5000
|
| 65 |
+
max_checkpoints_to_keep: 10
|
| 66 |
+
load_from: /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
|
| 67 |
+
evaluation:
|
| 68 |
+
every_train_iters: null
|
| 69 |
+
runtime_battle_mode: eval_mode
|
| 70 |
+
opponent_regime: meaningful_fixed_bot
|
| 71 |
+
opponent_impl: line_goal
|
| 72 |
+
primary_metric: win_rate_vs_fixed_bot
|
| 73 |
+
bot_strength_tier: moderate
|
| 74 |
+
bot_deterministic: true
|
| 75 |
+
requires_paired_audit: false
|
| 76 |
+
meaningful_fixed_bot: true
|
| 77 |
+
opponent_type: env_bot_and_previous_best_checkpoint
|
| 78 |
+
previous_best_checkpoint:
|
| 79 |
+
selector: best
|
| 80 |
+
update_policy: on_new_best
|
| 81 |
+
promotion_threshold: 0.0
|
| 82 |
+
fallback_to_env_bot: false
|
| 83 |
+
num_simulations: 10
|
| 84 |
+
evaluator_env_num: 10
|
| 85 |
+
n_evaluator_episode: 10
|
| 86 |
+
every_env_steps: 5000
|
| 87 |
+
env:
|
| 88 |
+
battle_mode: self_play_mode
|
| 89 |
+
battle_mode_in_simulation_env: self_play_mode
|
| 90 |
+
bot_action_type: rule
|
| 91 |
+
prob_random_action_in_bot: 0.0
|
| 92 |
+
collector_env_num: 4
|
| 93 |
+
evaluator_env_num: 10
|
| 94 |
+
n_evaluator_episode: 10
|
| 95 |
+
extra_config:
|
| 96 |
+
eval_random_start_enabled: true
|
| 97 |
+
eval_random_start_min_moves: 0
|
| 98 |
+
eval_random_start_max_moves: 20
|
| 99 |
+
eval_random_start_even_ply: true
|
| 100 |
+
stop_value: 2
|
| 101 |
+
defaults:
|
| 102 |
+
seed: 0
|
| 103 |
+
num_simulations: 50
|
| 104 |
+
batch_size: 256
|
| 105 |
+
update_per_collect: 50
|
| 106 |
+
learning_rate: 0.003
|
| 107 |
+
replay_buffer_size: 100000
|
| 108 |
+
discount_factor: 1
|
| 109 |
+
game_segment_length: 81
|
| 110 |
+
td_steps: 81
|
| 111 |
+
reanalyze_ratio: 0.0
|
| 112 |
+
num_unroll_steps: 5
|
| 113 |
+
piecewise_decay_lr_scheduler: false
|
| 114 |
+
max_env_step: 300000
|
| 115 |
+
model:
|
| 116 |
+
num_res_blocks: 1
|
| 117 |
+
num_channels: 64
|
| 118 |
+
model_info:
|
| 119 |
+
policy_type: efficientzero
|
| 120 |
+
model_class: EfficientZeroModel
|
| 121 |
+
trainable_params: 12626451
|
| 122 |
+
model_config:
|
| 123 |
+
observation_shape:
|
| 124 |
+
- 3
|
| 125 |
+
- 9
|
| 126 |
+
- 9
|
| 127 |
+
action_space_size: 81
|
| 128 |
+
image_channel: 3
|
| 129 |
+
num_res_blocks: 1
|
| 130 |
+
num_channels: 64
|
| 131 |
+
num_res_blocks: 1
|
| 132 |
+
num_channels: 64
|
| 133 |
+
observation_shape:
|
| 134 |
+
- 3
|
| 135 |
+
- 9
|
| 136 |
+
- 9
|
| 137 |
+
action_space_size: 81
|
| 138 |
+
image_channel: 3
|
| 139 |
+
resolved_main_config:
|
| 140 |
+
exp_name: /workspace/combinatorial_reasoning_post_training/models/simplified5_two_machine_continuation_20260520/shard-0/three_six_nine/attempt-03_260520_145623
|
| 141 |
+
env:
|
| 142 |
+
battle_mode: self_play_mode
|
| 143 |
+
bot_action_type: rule
|
| 144 |
+
channel_last: false
|
| 145 |
+
collector_env_num: 4
|
| 146 |
+
evaluator_env_num: 10
|
| 147 |
+
n_evaluator_episode: 10
|
| 148 |
+
manager:
|
| 149 |
+
shared_memory: false
|
| 150 |
+
battle_mode_in_simulation_env: self_play_mode
|
| 151 |
+
prob_random_action_in_bot: 0.0
|
| 152 |
+
eval_random_start_enabled: true
|
| 153 |
+
eval_random_start_min_moves: 0
|
| 154 |
+
eval_random_start_max_moves: 20
|
| 155 |
+
eval_random_start_even_ply: true
|
| 156 |
+
policy:
|
| 157 |
+
model:
|
| 158 |
+
observation_shape:
|
| 159 |
+
- 3
|
| 160 |
+
- 9
|
| 161 |
+
- 9
|
| 162 |
+
action_space_size: 81
|
| 163 |
+
image_channel: 3
|
| 164 |
+
num_res_blocks: 1
|
| 165 |
+
num_channels: 64
|
| 166 |
+
model_path: /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
|
| 167 |
+
cuda: true
|
| 168 |
+
env_type: board_games
|
| 169 |
+
action_type: varied_action_space
|
| 170 |
+
game_segment_length: 81
|
| 171 |
+
update_per_collect: 50
|
| 172 |
+
batch_size: 256
|
| 173 |
+
optim_type: Adam
|
| 174 |
+
learning_rate: 0.003
|
| 175 |
+
grad_clip_value: 0.5
|
| 176 |
+
num_simulations: 50
|
| 177 |
+
reanalyze_ratio: 0.0
|
| 178 |
+
td_steps: 81
|
| 179 |
+
discount_factor: 1
|
| 180 |
+
n_episode: 4
|
| 181 |
+
eval_freq: 300001
|
| 182 |
+
replay_buffer_size: 100000
|
| 183 |
+
collector_env_num: 4
|
| 184 |
+
evaluator_env_num: 10
|
| 185 |
+
use_wandb: true
|
| 186 |
+
best_ckpt_strategy: raw
|
| 187 |
+
best_ckpt_ema_alpha: 0.3
|
| 188 |
+
best_ckpt_min_episodes: 20
|
| 189 |
+
num_unroll_steps: 5
|
| 190 |
+
battle_mode: self_play_mode
|
| 191 |
+
battle_mode_in_simulation_env: self_play_mode
|
| 192 |
+
piecewise_decay_lr_scheduler: false
|
| 193 |
+
mcts_ctree: true
|
| 194 |
+
eval_opponent_type: env_bot_and_previous_best_checkpoint
|
| 195 |
+
previous_best_checkpoint:
|
| 196 |
+
path: null
|
| 197 |
+
selector: best
|
| 198 |
+
update_policy: on_new_best
|
| 199 |
+
num_simulations: 10
|
| 200 |
+
n_evaluator_episode: 10
|
| 201 |
+
evaluator_env_num: 10
|
| 202 |
+
promotion_threshold: 0.0
|
| 203 |
+
fallback_to_env_bot: false
|
| 204 |
+
device: cuda
|
| 205 |
+
resolved_create_config:
|
| 206 |
+
env:
|
| 207 |
+
type: three_six_nine
|
| 208 |
+
import_names:
|
| 209 |
+
- custom_games.three_six_nine.envs.three_six_nine_env
|
| 210 |
+
env_manager:
|
| 211 |
+
type: subprocess
|
| 212 |
+
policy:
|
| 213 |
+
type: efficientzero
|
| 214 |
+
import_names:
|
| 215 |
+
- lzero.policy.efficientzero
|