duoduoyeah commited on
Commit
e866845
·
verified ·
1 Parent(s): ffeef56

Add files using upload-large-folder tool

Browse files
Files changed (3) hide show
  1. meta_000600.json +50 -0
  2. model_000600.pt +3 -0
  3. optim_000600_rank0.pt +3 -0
meta_000600.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "step": 600,
3
+ "val_bpb": 0.6578793481726156,
4
+ "model_config": {
5
+ "sequence_len": 1024,
6
+ "vocab_size": 4096,
7
+ "n_layer": 4,
8
+ "n_head": 2,
9
+ "n_kv_head": 2,
10
+ "n_embd": 256
11
+ },
12
+ "user_config": {
13
+ "run": "dummy",
14
+ "device_type": "",
15
+ "depth": 4,
16
+ "max_seq_len": 1024,
17
+ "num_iterations": -1,
18
+ "target_flops": -1.0,
19
+ "target_param_data_ratio": 60,
20
+ "device_batch_size": 256,
21
+ "total_batch_size": 524288,
22
+ "embedding_lr": 0.2,
23
+ "unembedding_lr": 0.004,
24
+ "weight_decay": 0.0,
25
+ "matrix_lr": 0.02,
26
+ "grad_clip": 1.0,
27
+ "warmup_ratio": 0.0,
28
+ "warmdown_ratio": 0.2,
29
+ "final_lr_frac": 0.0,
30
+ "resume_from_step": -1,
31
+ "eval_every": 250,
32
+ "eval_tokens": 10485760,
33
+ "core_metric_every": 2000,
34
+ "core_metric_max_per_task": 500,
35
+ "sample_every": 2000,
36
+ "save_every": -1,
37
+ "model_tag": ""
38
+ },
39
+ "device_batch_size": 256,
40
+ "max_seq_len": 1024,
41
+ "dataloader_state_dict": {
42
+ "pq_idx": 5,
43
+ "rg_idx": 35
44
+ },
45
+ "loop_state": {
46
+ "min_val_bpb": 0.6578793481726156,
47
+ "smooth_train_loss": 1.924130117203954,
48
+ "total_training_time": 174.06030464172363
49
+ }
50
+ }
model_000600.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b90b59724f29e7ff517a484339c8573d8391642e155d2fc288686a047af01471
3
+ size 18885365
optim_000600_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68de462e5b9abebce7726fdadddc0e0b56ce4b5fd78b1fe69d0fef82992abf96
3
+ size 25176405