Checkpoint iteration 2

runs/policy_spatial_v15_3/metadata_iter_002.json

@@ -0,0 +1,162 @@
+{
+  "iteration": 2,
+  "timestamp": "2026-05-30T03:40:15.858917+00:00",
+  "buffer_size": 942,
+  "config": {
+    "iterations": 600,
+    "episodes_per_iter": 18,
+    "mcts_sims": 800,
+    "c_puct": 1.5,
+    "temp_threshold": 8,
+    "add_noise": true,
+    "seed": 42,
+    "verbose_logs": true,
+    "episode_log_every": 25,
+    "epochs": 1,
+    "batch_size": 256,
+    "learning_rate": 0.0001,
+    "weight_decay": 0.0001,
+    "value_loss_coeff": 1.0,
+    "reward_shaping_enabled": false,
+    "reward_shaping_scale": 0.1,
+    "reward_shaping_gamma": 0.99,
+    "reward_shaping_material_weight": 0.6,
+    "reward_shaping_mobility_weight": 0.4,
+    "reward_shaping_draw_penalty": 0.1,
+    "league_enabled": true,
+    "league_path": "checkpoints/league_ratings.json",
+    "league_checkpoint_pool_top_k": 3,
+    "league_checkpoint_pool_min_games": 4,
+    "league_selfplay_checkpoint_prob": 0.55,
+    "league_champion_games": 6,
+    "buffer_size": 200000,
+    "val_split": 0.1,
+    "shuffle_train_val_split": true,
+    "train_recent_fraction": 0.3,
+    "train_recent_window_fraction": 0.4,
+    "d_model": 384,
+    "nhead": 6,
+    "num_layers": 12,
+    "dim_feedforward": 768,
+    "dropout": 0.0,
+    "log_dir": "/kaggle/working/logs",
+    "checkpoint_dir": "/kaggle/working/checkpoints",
+    "save_every": 2,
+    "keep_last_n_local_checkpoints": 1,
+    "keep_last_n_log_versions": 2,
+    "keep_last_n_hf_checkpoints": 1,
+    "onnx_path": "ataxx_model.onnx",
+    "export_onnx": false,
+    "hf_enabled": true,
+    "hf_repo_id": "dieg0code/ataxx-zero",
+    "hf_run_id": "policy_spatial_v15_3",
+    "hf_bootstrap_run_id": "",
+    "hf_reset_iteration": false,
+    "hf_token_env": "HF_TOKEN",
+    "hf_local_dir": "/kaggle/working/hf_checkpoints",
+    "wandb_enabled": true,
+    "wandb_project": "ataxx-zero",
+    "wandb_entity": "dieg0code-ai",
+    "wandb_run_name": "policy_spatial_v15_3",
+    "max_pending_hf_uploads": 12,
+    "hf_upload_future_timeout_s": 900.0,
+    "show_progress_bar": false,
+    "trainer_log_every_n_steps": 99999,
+    "monitor_log_every": 5,
+    "epoch_pulse_every": 2,
+    "num_workers": 2,
+    "persistent_workers": true,
+    "strict_probs": false,
+    "trainer_devices": 2,
+    "trainer_strategy": "ddp_spawn",
+    "trainer_precision": "16-mixed",
+    "trainer_benchmark": true,
+    "ddp_timeout_seconds": 180,
+    "mcts_use_amp": true,
+    "mcts_cache_size": 100000,
+    "mcts_leaf_batch_size": 32,
+    "mcts_dirichlet_alpha": 0.3,
+    "mcts_dirichlet_frac": 0.25,
+    "fail_on_selfplay_parallel_error": false,
+    "fail_on_hf_upload_error": false,
+    "opponent_self_prob": 0.45,
+    "opponent_heuristic_prob": 0.0,
+    "opponent_random_prob": 0.05,
+    "opponent_heuristic_level": "normal",
+    "opponent_heuristic_epsilon": 0.0,
+    "opponent_heuristic_easy_prob": 0.2,
+    "opponent_heuristic_normal_prob": 0.5,
+    "opponent_heuristic_hard_prob": 0.3,
+    "model_side_swap_prob": 0.5,
+    "eval_enabled": true,
+    "eval_every": 8,
+    "eval_games": 40,
+    "eval_sims": 256,
+    "eval_heuristic_level": "hard",
+    "eval_heuristic_levels": "easy,normal,hard,apex,gambit,sentinel",
+    "restore_best_on_regression": false,
+    "eval_regression_delta": 0.06,
+    "eval_regression_patience": 2,
+    "baseline_checkpoint": "liga",
+    "baseline_hf_run_id": "policy_spatial_v8",
+    "baseline_composite": 0.85,
+    "baseline_h2h_min_score": 0.25,
+    "eval_absolute_patience": 2,
+    "eval_absolute_delta": 0.03,
+    "eval_absolute_action": "abort",
+    "eval_absolute_abort_mode": "h2h",
+    "eval_absolute_min_iteration": 150,
+    "selfplay_workers": 2,
+    "selfplay_progress_every_s": 120.0,
+    "selfplay_episode_timeout_s": 1800.0,
+    "compile_model": false,
+    "quiet_mode": false,
+    "warmup_games": 0,
+    "warmup_epochs": 0,
+    "warmup_heuristic_level": "sentinel",
+    "warmup_heuristic_levels": "hard,apex,sentinel",
+    "pretrain_dataset_path": "",
+    "pretrain_epochs": 0,
+    "human_replay_path": "",
+    "human_batch_fraction": 0.0,
+    "human_value_mask": false,
+    "symmetry_augmentation": true,
+    "value_head_depth": 1,
+    "count_head_enabled": false,
+    "count_loss_coeff": 0.0,
+    "eval_composite_uses_h2h_only": true,
+    "transformer_pre_ln": true,
+    "mcts_fpu_reduction": 0.25,
+    "mcts_virtual_loss": 1.0,
+    "pos_embed_2d": true,
+    "lr_warmup_steps": 0,
+    "adam_beta2": 0.95,
+    "ema_decay": 0.999,
+    "value_mcts_q_lambda": 0.25,
+    "max_half_moves_per_episode": 120,
+    "random_opening_prob": 0.25,
+    "random_opening_max_plies": 3,
+    "mcts_prior_uniform_mix": 0.2,
+    "pcr_enabled": true,
+    "pcr_fast_sims": 100,
+    "pcr_full_sim_prob": 0.25,
+    "resign_enabled": true,
+    "resign_value_threshold": -0.95,
+    "resign_consecutive_plies": 8,
+    "resign_min_plies": 20,
+    "resign_audit_prob": 0.1,
+    "mcts_forced_playout_k": 2.0,
+    "mcts_policy_target_prune_forced": true,
+    "patch_embed_conv": true
+  },
+  "stats": {
+    "replay_size": 942,
+    "best_eval_score": -1.0,
+    "train_loss_total": 5.244462966918945,
+    "train_loss_value": 1.4368581771850586,
+    "train_loss_policy": 3.807605028152466,
+    "train_value_mae": 0.909564733505249,
+    "train_policy_accuracy": 0.03537735715508461,
+    "train_lr": 9.99993208097294e-05
+  }
+}