Instructions to use TehranNLP-org/electra-base-sst2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use TehranNLP-org/electra-base-sst2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="TehranNLP-org/electra-base-sst2")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("TehranNLP-org/electra-base-sst2") model = AutoModelForSequenceClassification.from_pretrained("TehranNLP-org/electra-base-sst2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "model_name_or_path": "google/electra-base-discriminator", | |
| "task_name": "sst2", | |
| "output_dir": "./output/electra/sst2/0001/SEED0042/", | |
| "learning_rate": 3e-05, | |
| "num_train_epochs": 3, | |
| "per_device_eval_batch_size": 32, | |
| "per_device_train_batch_size": 32, | |
| "seed": 42, | |
| "warmup_steps": 2000, | |
| "do_train": true, | |
| "do_eval": true, | |
| "do_predict": true, | |
| "pad_to_max_length": false, | |
| "max_seq_length": 64, | |
| "report_to": [], | |
| "save_strategy": "epoch", | |
| "evaluation_strategy": "epoch", | |
| "logging_steps": 2500, | |
| "use_fast_tokenizer": true, | |
| "group_by_length": true, | |
| "save_training_dynamics": true, | |
| "save_training_dynamics_after_epoch": true | |
| } |