logasanjeev's picture
Upload config.json for updated GoEmotions BERT model
09420d9 verified
Raw History Blame
195 Bytes
{"model_name": "bert-base-uncased", "num_labels": 28, "batch_size": 16, "max_length": 128, "epochs": 5, "learning_rate": 2e-05, "weight_decay": 0.01, "focal_loss_alpha": 1, "focal_loss_gamma": 2}