| language: en | |
| license: mit | |
| datasets: | |
| - jigsaw-toxic-comment-classification-challenge | |
| - kaggle/jigsaw-toxic-comment-classification-challenge | |
| tags: | |
| - text-classification | |
| base_model: FacebookAI/roberta-base | |
| library_name: peft | |
| # Toxicity Detection Model | |
| train_runtime': 501.1294 | |
| r = 4 | |
| num_train_epochs=1 | |
| optim="adamw_torch_fused", | |
| lr_scheduler_type="cosine", | |
| metric_for_best_model="f1", | |
| ### Framework versions | |
| - PEFT 0.14.0 |