valurank
/

finetuned-distilbert-adult-content-detection

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

abdulmatinomotoso commited on Jun 22, 2022

Commit

7b43e2c

•

1 Parent(s): 40ef089

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ The following hyperparameters were used during training:
 - optimizer: AdamW(lr=1e-5 and epsilon=1e-08)
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 0
-- num_epochs: 5
 ### Training results
 | Training Loss | Epoch |  Validation Loss | f1 score   |
 |:-------------:|:-----:|:---------------: |:------:|

 - optimizer: AdamW(lr=1e-5 and epsilon=1e-08)
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 0
+- num_epochs: 2
 ### Training results
 | Training Loss | Epoch |  Validation Loss | f1 score   |
 |:-------------:|:-----:|:---------------: |:------:|