File size: 5,237 Bytes
7209010 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 | {
"iteration": 34,
"timestamp": "2026-05-31T03:11:35.253495+00:00",
"buffer_size": 56045,
"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": 56045,
"best_eval_score": 0.9874999999999999,
"train_loss_total": 2.0576391220092773,
"train_loss_value": 0.2748979330062866,
"train_loss_policy": 1.7827417850494385,
"train_value_mae": 0.37447378039360046,
"train_policy_accuracy": 0.46905356645584106,
"train_lr": 9.99993208097294e-05
}
} |