learning_rate: 2e-05
train_batch_size: 8
eval_batch_size: 8
seed: 42
gradient_accumulation_steps: 2
total_train_batch_size: 16
optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: cosine
lr_scheduler_warmup_ratio: 0.2
num_epochs: 6
mixed_precision_training: Native AMP

Training Loss	Epoch	Step	Validation Loss	Accuracy	Precision	Recall	F1
3.1502	1.0	10452	2.4660	0.5884	0.6502	0.5884	0.6030
2.39	2.0	20904	2.3147	0.6235	0.6775	0.6235	0.6339
2.1944	3.0	31356	2.2647	0.6450	0.6954	0.6450	0.6568
2.0624	4.0	41808	2.2534	0.6560	0.7023	0.6560	0.6679
1.9719	5.0	52260	2.2555	0.6609	0.7019	0.6609	0.6711
1.924	6.0	62712	2.2599	0.6637	0.7024	0.6637	0.6732

AbrarAbhinaya/DistilbertScenario6-news-classifier

DistilbertScenario6-news-classifier