nicholasKluge
/

ToxicityModel

Text Classification

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Nov 4, 2023

Commit

1e88940

•

1 Parent(s): 64be6dc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ The model was trained with a dataset composed of `toxic_response` and `non_toxic
 - **Dataset:** [Toxic-Aira Dataset](https://huggingface.co/datasets/nicholasKluge/toxic-aira-dataset)
 - **Language:** English
 - **Number of Training Steps:** 1000
-- **Batch size:** 42
 - **Optimizer:** `torch.optim.AdamW`
 - **Learning Rate:** 5e-5
 - **GPU:** 1 NVIDIA A100-SXM4-40GB

 - **Dataset:** [Toxic-Aira Dataset](https://huggingface.co/datasets/nicholasKluge/toxic-aira-dataset)
 - **Language:** English
 - **Number of Training Steps:** 1000
+- **Batch size:** 32
 - **Optimizer:** `torch.optim.AdamW`
 - **Learning Rate:** 5e-5
 - **GPU:** 1 NVIDIA A100-SXM4-40GB