SAVRN
Search Contact SAVRN

Independent publisher

Leandro von Werra

lvwerra

NLP and RL

Models in Library1
Datasets in Library0
Models on Hugging Face46
Followers885

Models

Model · Text classification

distilbert-imdb

Leandro von Werra

This model is a fine-tuned version of distilbert-base-uncased on the imdb dataset (training notebook is here). It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 5e-05 - trainbatchsize: 16 - evalbatchsize: 16 - lrschedulertype: linear - numepochs: 1 - Transformers 4.15.0 - Pytorch 1.10.0+cu111 - Datasets 1.17.0 - Tokenizers 0.10.3

Open weights apache-2.0 512 tokens transformers