leminhhung0101 commited on
Commit
7678c2c
·
verified ·
1 Parent(s): 1e4ed09

Upload 2 files

Browse files
Files changed (2) hide show
  1. run_config.json +77 -0
  2. train_metrics.json +18 -0
run_config.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_path": "/kaggle/input/datasets/leminhhung0101/qwen3-4b-thinking-2507",
3
+ "data_root": "/kaggle/input/datasets/leminhhung0101/arc-qlora/arc_qlora_dataset",
4
+ "output_dir": "/kaggle/working/arc_qwen3_4b_thinking_2507_qlora_resume",
5
+ "resume_checkpoint": "/kaggle/input/datasets/leminhhung0101/arc-qlora/checkpoint-4000",
6
+ "episode_counts": {
7
+ "arc2024_training": 10000,
8
+ "arc2024_eval_as_train": 10000,
9
+ "arc_agi_2_training": 25000,
10
+ "neurogolf_arcgen": 40000
11
+ },
12
+ "locked_benchmark": "/kaggle/input/datasets/leminhhung0101/arc-qlora/arc_qlora_dataset/benchmark/arc_agi_2_evaluation",
13
+ "max_length": 8192,
14
+ "runtime_max_tokens": 5632,
15
+ "runtime_truncated_episodes": 1670,
16
+ "target_truncation": "never",
17
+ "max_demos": 3,
18
+ "color_permute_prob": 0.6,
19
+ "token_cache": "/kaggle/input/datasets/leminhhung0101/arc-qlora/arc_qlora_token_cache/3384cef794411770",
20
+ "cache_stats": {
21
+ "config": {
22
+ "model_path": "/kaggle/input/datasets/leminhhung0101/qwen3-4b-thinking-2507",
23
+ "data_root": "/kaggle/input/datasets/leminhhung0101/arc-qlora/arc_qlora_dataset",
24
+ "episodes_arc2024_train": 10000,
25
+ "episodes_arc2024_eval": 10000,
26
+ "episodes_arc2_train": 25000,
27
+ "episodes_neurogolf": 40000,
28
+ "seed": 42,
29
+ "max_length": 8192,
30
+ "max_demos": 3,
31
+ "color_permute_prob": 0.6,
32
+ "cache_format": 1
33
+ },
34
+ "episodes": 85000,
35
+ "total_tokens": 117896318,
36
+ "mean_length": 1387.015505882353,
37
+ "length_percentiles": {
38
+ "50": 917.0,
39
+ "75": 1853.0,
40
+ "90": 3061.0,
41
+ "95": 4021.2000000000116,
42
+ "99": 7115.0,
43
+ "99.5": 7477.0,
44
+ "100": 7973.0
45
+ },
46
+ "build_seconds": 297.9708034992218,
47
+ "dtype": "int32"
48
+ },
49
+ "lora": {
50
+ "r": 64,
51
+ "alpha": 128,
52
+ "dropout": 0.0,
53
+ "targets": [
54
+ "q_proj",
55
+ "k_proj",
56
+ "v_proj",
57
+ "o_proj",
58
+ "gate_proj",
59
+ "up_proj",
60
+ "down_proj"
61
+ ]
62
+ },
63
+ "training": {
64
+ "learning_rate": 0.0001,
65
+ "optimizer": "adamw_bnb_8bit",
66
+ "epochs": 1.0,
67
+ "per_device_batch_size": 1,
68
+ "gradient_accumulation": 4,
69
+ "world_size": 4,
70
+ "effective_batch_size": 16,
71
+ "gradient_checkpointing": true,
72
+ "checkpointing_mode": "on",
73
+ "attention": "sdpa",
74
+ "group_by_length": false,
75
+ "precision": "NF4 4-bit base + BF16 compute"
76
+ }
77
+ }
train_metrics.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_runtime": 19106.7506,
3
+ "train_samples_per_second": 4.449,
4
+ "train_steps_per_second": 0.278,
5
+ "total_flos": 2.637603248881533e+18,
6
+ "train_loss": 0.006949852055207972,
7
+ "epoch": 0.9999058823529412,
8
+ "wall_time_seconds": 19115.41075849533,
9
+ "world_size": 4,
10
+ "effective_batch_size": 16,
11
+ "total_episodes": 85000,
12
+ "token_cache": "/kaggle/input/datasets/leminhhung0101/arc-qlora/arc_qlora_token_cache/3384cef794411770",
13
+ "gradient_checkpointing": true,
14
+ "attn_implementation": "sdpa",
15
+ "runtime_max_tokens": 5632,
16
+ "runtime_truncated_episodes": 1670,
17
+ "resume_checkpoint": "/kaggle/input/datasets/leminhhung0101/arc-qlora/checkpoint-4000"
18
+ }