| { | |
| "architectures": ["BertForSequenceClassification"], | |
| "model_type": "bert", | |
| "pipeline_tag": "text-classification", | |
| "dataset_path": "/raid/Toxic_telugu_dataset/train/labeled_dataset.csv", | |
| "model_name": "google-bert/bert-base-multilingual-uncased", | |
| "batch_size": 16, | |
| "epochs": 3, | |
| "max_length": 128, | |
| "learning_rate": 2e-05, | |
| "accumulation_steps": 2, | |
| "embedding_dim": 768 | |
| } |