learning_rate: 0.0018
train_batch_size: 32
eval_batch_size: 32
seed: 42
optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.95) and epsilon=1e-06 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: cosine
lr_scheduler_warmup_steps: 0.05
num_epochs: 1.0

Training Loss	Epoch	Step	Validation Loss
No log	0	0	6.9367
2.5547	0.0500	2014	2.5635
2.3775	0.1000	4028	2.3864
2.3124	0.1500	6042	2.3258
2.2757	0.2000	8056	2.2938
2.2581	0.2500	10070	2.2669
2.2369	0.3000	12084	2.2510
2.2110	0.3500	14098	2.2340
2.1943	0.4000	16112	2.2203
2.1867	0.4500	18126	2.2001
2.1623	0.5000	20140	2.1832
2.1521	0.5500	22154	2.1676
2.1146	0.6000	24168	2.1447
2.1124	0.6500	26182	2.1262
2.0921	0.7000	28196	2.1067
2.0619	0.7500	30210	2.0862
2.0472	0.8000	32224	2.0700
2.0384	0.8500	34238	2.0552
2.0219	0.9000	36252	2.0438
2.0146	0.9500	38266	2.0383
2.0082	1.0	40278	2.0368

stage-babylm/llama-128-2L

llama-128-2L