bd4sur commited on
Commit
ec22b64
·
verified ·
1 Parent(s): b10b36d

Upload config_nano_168m_625000_sft_875000_amateur_radio_890000.json

Browse files
config_nano_168m_625000_sft_875000_amateur_radio_890000.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "use_lora": false,
3
+ "lora_rank": 8,
4
+ "lora_alpha": 16,
5
+ "lora_dropout": 0.0,
6
+
7
+ "from_checkpoint": "/home/bd4sur/ai/Nano/checkpoint/checkpoint_20250104_120620_step_875000.pt",
8
+ "save_checkpoint_to": "/home/bd4sur/ai/Nano/checkpoint",
9
+ "dataset_path": [
10
+ ["/home/bd4sur/ai/Nano/dataset_preprocessed/sft_train.base64", "/home/bd4sur/ai/Nano/dataset_preprocessed/sft_val.base64"]
11
+ ],
12
+ "tokenizer_path": "/home/bd4sur/ai/Nano/tokenizer/tokenizer_16384.json",
13
+
14
+ "random_seed": 39,
15
+ "batch_size": 10,
16
+ "gradient_accumulation_steps": 1,
17
+ "grad_clip": 1.0,
18
+
19
+ "dropout": 0.1,
20
+
21
+ "learning_rate": 5e-6,
22
+ "weight_decay": 1e-1,
23
+ "beta1": 0.9,
24
+ "beta2": 0.95,
25
+
26
+ "decay_lr": false,
27
+ "warmup_iters": 10000,
28
+ "lr_decay_iters": 1e9,
29
+ "min_lr": 1e-8,
30
+
31
+ "eval_interval": 500,
32
+ "log_interval": 10,
33
+ "eval_iters": 5,
34
+
35
+ "backend": "nccl",
36
+ "device": "cuda",
37
+ "sdp_kernel": "flash",
38
+ "dtype": "bfloat16",
39
+ "use_amp": true
40
+ }