Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

arnabdhar
/
tinybert-imdb

Text Classification
Transformers
TensorBoard
ONNX
Safetensors
PyTorch
English
bert
movie-review-sentiment
BertForSequenceClassification
Generated from Trainer
Eval Results (legacy)
text-embeddings-inference
Model card Files Files and versions
xet
Metrics Training metrics Community

Instructions to use arnabdhar/tinybert-imdb with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use arnabdhar/tinybert-imdb with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="arnabdhar/tinybert-imdb")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("arnabdhar/tinybert-imdb")
    model = AutoModelForSequenceClassification.from_pretrained("arnabdhar/tinybert-imdb")
  • Notebooks
  • Google Colab
  • Kaggle
tinybert-imdb
36.1 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 10 commits
arnabdhar's picture
arnabdhar
Updated README.md
b12e790 unverified over 2 years ago
  • runs
    End of training over 2 years ago
  • .gitattributes
    1.52 kB
    initial commit over 2 years ago
  • README.md
    4.4 kB
    Updated README.md over 2 years ago
  • config.json
    815 Bytes
    Upload config.json with huggingface_hub over 2 years ago
  • model.onnx
    17.6 MB
    xet
    Upload model.onnx with huggingface_hub over 2 years ago
  • model.safetensors
    17.5 MB
    xet
    End of training over 2 years ago
  • special_tokens_map.json
    695 Bytes
    Upload special_tokens_map.json with huggingface_hub over 2 years ago
  • tokenizer.json
    712 kB
    End of training over 2 years ago
  • tokenizer_config.json
    1.46 kB
    Upload tokenizer_config.json with huggingface_hub over 2 years ago
  • training_args.bin

    Detected Pickle imports (8)

    • "transformers.trainer_utils.IntervalStrategy",
    • "transformers.training_args.OptimizerNames",
    • "accelerate.utils.dataclasses.DistributedType",
    • "accelerate.state.PartialState",
    • "transformers.trainer_utils.SchedulerType",
    • "transformers.training_args.TrainingArguments",
    • "torch.device",
    • "transformers.trainer_utils.HubStrategy"

    How to fix it?

    4.6 kB
    xet
    End of training over 2 years ago
  • vocab.txt
    232 kB
    End of training over 2 years ago