AdamCodd
/

distilbert-base-uncased-finetuned-emotion-balanced

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

AdamCodd commited on Sep 9, 2023

Commit

1766d87

•

1 Parent(s): a9e4130

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -65,7 +65,7 @@ The following hyperparameters were used during training:
 - train_batch_size: 32
 - eval_batch_size: 64
 - seed: 1270
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 150
 - num_epochs: 1

 - train_batch_size: 32
 - eval_batch_size: 64
 - seed: 1270
+- optimizer: AdamW with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 150
 - num_epochs: 1