vibhorag101
/

roberta-base-suicide-prediction-phr-v2

Text Classification

Generated from Trainer

Model card Files Files and versions

vibhorag101 commited on Mar 16, 2024

Commit

de48637

·

verified ·

1 Parent(s): 28c4143

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -74,7 +74,8 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.06
 - num_epochs: 3
 - eval_steps: 500
 - save_steps: 500

 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
+- weight_decay=0.1
+- warmup_ratio: 0.06
 - num_epochs: 3
 - eval_steps: 500
 - save_steps: 500