search

unitaryai / detoxify

Trained models & code to predict toxic comments on all 3 Jigsaw Toxic Comment Challenges. Built using ⚡ Pytorch Lightning and 🤗 Transformers. For access to our API, please email us at contact@unitary.ai.
https://www.unitary.ai/
Apache License 2.0
893 stars 115 forks source link

Number of epochs to get the best model #72

Open mathuryash5 opened 1 year ago

mathuryash5 commented 1 year ago

Hello,

I wanted to reproduce the results by the models and was wondering the number of epochs each model had to be trained to get the scores shown.

Thank you!

laurahanu commented 1 year ago

Hello,

For most of the models it's around 2-3 epochs!

voarsh2 commented 5 months ago

I find it strange the default was set to 100 - and you recommend 2/3? How many epochs did you run when you make your checkpoints public - that would be nice to know