dieg0code commited on
Commit
fbde9f4
·
verified ·
1 Parent(s): d23ca59

Checkpoint iteration 4

Browse files
runs/policy_spatial_v3/metadata_iter_004.json ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "iteration": 4,
3
+ "timestamp": "2026-03-08T04:55:49.607317+00:00",
4
+ "buffer_size": 3196,
5
+ "config": {
6
+ "iterations": 220,
7
+ "episodes_per_iter": 20,
8
+ "mcts_sims": 160,
9
+ "c_puct": 1.5,
10
+ "temp_threshold": 28,
11
+ "add_noise": true,
12
+ "seed": 42,
13
+ "verbose_logs": true,
14
+ "episode_log_every": 25,
15
+ "epochs": 2,
16
+ "batch_size": 224,
17
+ "learning_rate": 0.0003,
18
+ "weight_decay": 0.0001,
19
+ "value_loss_coeff": 0.5,
20
+ "buffer_size": 50000,
21
+ "val_split": 0.1,
22
+ "shuffle_train_val_split": true,
23
+ "train_recent_fraction": 0.7,
24
+ "train_recent_window_fraction": 0.4,
25
+ "d_model": 128,
26
+ "nhead": 8,
27
+ "num_layers": 6,
28
+ "dim_feedforward": 512,
29
+ "dropout": 0.1,
30
+ "log_dir": "/kaggle/working/logs",
31
+ "checkpoint_dir": "/kaggle/working/checkpoints",
32
+ "save_every": 1,
33
+ "keep_last_n_local_checkpoints": 256,
34
+ "keep_last_n_log_versions": 2,
35
+ "keep_last_n_hf_checkpoints": 3,
36
+ "onnx_path": "ataxx_model.onnx",
37
+ "export_onnx": false,
38
+ "hf_enabled": true,
39
+ "hf_repo_id": "dieg0code/ataxx-zero",
40
+ "hf_run_id": "policy_spatial_v3",
41
+ "hf_bootstrap_run_id": "policy_spatial_v2",
42
+ "hf_reset_iteration": true,
43
+ "hf_token_env": "HF_TOKEN",
44
+ "hf_local_dir": "hf_checkpoints",
45
+ "max_pending_hf_uploads": 6,
46
+ "hf_upload_future_timeout_s": 900.0,
47
+ "show_progress_bar": false,
48
+ "trainer_log_every_n_steps": 99999,
49
+ "monitor_log_every": 1,
50
+ "epoch_pulse_every": 1,
51
+ "num_workers": 2,
52
+ "persistent_workers": true,
53
+ "strict_probs": false,
54
+ "trainer_devices": 2,
55
+ "trainer_strategy": "ddp_spawn",
56
+ "trainer_precision": "16-mixed",
57
+ "trainer_benchmark": true,
58
+ "ddp_timeout_seconds": 120,
59
+ "mcts_use_amp": true,
60
+ "mcts_cache_size": 100000,
61
+ "mcts_leaf_batch_size": 32,
62
+ "fail_on_selfplay_parallel_error": false,
63
+ "fail_on_hf_upload_error": false,
64
+ "opponent_self_prob": 0.45,
65
+ "opponent_heuristic_prob": 0.5,
66
+ "opponent_random_prob": 0.05,
67
+ "opponent_heuristic_level": "normal",
68
+ "opponent_heuristic_easy_prob": 0.2,
69
+ "opponent_heuristic_normal_prob": 0.5,
70
+ "opponent_heuristic_hard_prob": 0.3,
71
+ "model_side_swap_prob": 0.5,
72
+ "eval_enabled": true,
73
+ "eval_every": 6,
74
+ "eval_games": 12,
75
+ "eval_sims": 160,
76
+ "eval_heuristic_level": "hard",
77
+ "eval_heuristic_levels": "hard,apex,sentinel",
78
+ "restore_best_on_regression": true,
79
+ "eval_regression_delta": 0.03,
80
+ "eval_regression_patience": 2,
81
+ "selfplay_workers": 2,
82
+ "selfplay_progress_every_s": 45.0,
83
+ "selfplay_episode_timeout_s": 600.0,
84
+ "compile_model": false,
85
+ "quiet_mode": false,
86
+ "warmup_games": 240,
87
+ "warmup_epochs": 3,
88
+ "warmup_heuristic_level": "sentinel",
89
+ "warmup_heuristic_levels": "hard,apex,sentinel"
90
+ },
91
+ "stats": {
92
+ "replay_size": 3196,
93
+ "best_eval_score": -1.0
94
+ }
95
+ }