Hugging Face's logo

SlayerLab
/
bdh-scaling

PyTorch
English
Polish
Czech
bdh
linear-attention
scaling-laws
replication
Model card Files
xet
 Community
bdh-scaling
File size: 2,316 Bytes
2f957bf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
[
  {
    "step": 0,
    "val_loss": 5.724483394622803,
    "tok_per_s": 0.0
  },
  {
    "step": 0,
    "train_loss": 5.738528728485107,
    "lr": 1e-06,
    "grad_norm": 18.557702700432156
  },
  {
    "step": 400,
    "train_loss": 2.5503506660461426,
    "lr": 0.00040100000000000004,
    "grad_norm": 1.5563106453646307
  },
  {
    "step": 500,
    "val_loss": 4.181924271583557,
    "tok_per_s": 2184.0695229395146
  },
  {
    "step": 800,
    "train_loss": 2.169980049133301,
    "lr": 0.0008010000000000001,
    "grad_norm": 1.385430282630754
  },
  {
    "step": 1000,
    "val_loss": 3.9483901262283325,
    "tok_per_s": 3970.938096122552
  },
  {
    "step": 1200,
    "train_loss": 2.0457634925842285,
    "lr": 0.000988964669915361,
    "grad_norm": 1.0112889612874592
  },
  {
    "step": 1500,
    "val_loss": 3.940820741653442,
    "tok_per_s": 5463.298530973348
  },
  {
    "step": 1600,
    "train_loss": 1.9771353006362915,
    "lr": 0.0009042005118087267,
    "grad_norm": 0.8698389811850052
  },
  {
    "step": 2000,
    "val_loss": 3.6034568548202515,
    "tok_per_s": 6733.318830432771
  },
  {
    "step": 2000,
    "train_loss": 1.9552191495895386,
    "lr": 0.0007495464131639923,
    "grad_norm": 0.8105395679074034
  },
  {
    "step": 2400,
    "train_loss": 2.0290591716766357,
    "lr": 0.0005517434726252113,
    "grad_norm": 0.7772307357811391
  },
  {
    "step": 2500,
    "val_loss": 3.8027327179908754,
    "tok_per_s": 7825.7691054805655
  },
  {
    "step": 2800,
    "train_loss": 1.4810876846313477,
    "lr": 0.00034499361559483973,
    "grad_norm": 0.5642906609112506
  },
  {
    "step": 3000,
    "val_loss": 3.4605381011962892,
    "tok_per_s": 8775.98397199023
  },
  {
    "step": 3200,
    "train_loss": 1.6780955791473389,
    "lr": 0.00016504577061691466,
    "grad_norm": 0.6131905937913334
  },
  {
    "step": 3500,
    "val_loss": 3.523012411594391,
    "tok_per_s": 9610.316128988608
  },
  {
    "step": 3600,
    "train_loss": 1.8657923936843872,
    "lr": 4.301455486296946e-05,
    "grad_norm": 0.8052576967507755
  },
  {
    "step": 4000,
    "val_loss": 3.5965240597724915,
    "tok_per_s": 10347.765560997088
  },
  {
    "step": 4000,
    "train_loss": 1.5910844802856445,
    "lr": 2.741556527352529e-10,
    "grad_norm": 0.5769667369420309
  }
]