learning_rate: 0.0005
train_batch_size: 32
eval_batch_size: 32
seed: 42
gradient_accumulation_steps: 8
total_train_batch_size: 256
optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: cosine
lr_scheduler_warmup_steps: 1000
num_epochs: 1
mixed_precision_training: Native AMP

Training Loss	Epoch	Step	Validation Loss
2.5518	0.0766	5000	1.7167
1.6589	0.1533	10000	1.5073
1.52	0.2299	15000	1.4102
1.4458	0.3065	20000	1.3471
1.3886	0.3832	25000	1.2995
1.3392	0.4598	30000	1.2535
1.2947	0.5365	35000	1.2118
1.2495	0.6131	40000	1.1706
1.2092	0.6897	45000	1.1327
1.1721	0.7664	50000	1.0993
1.1456	0.8430	55000	1.0765
1.1243	0.9196	60000	1.0646
1.1171	0.9963	65000	1.0616

abhiramvad/codeparrot-ds

codeparrot-ds