config.json

{
    "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
}