config.json

{
  "model_name": "Magical-1 Sun",
  "model_type": "bert",
  "base_model": "bert-base-uncased",
  "num_labels": 2,
  "max_length": 128,
  "batch_size": 2,
  "learning_rate": 2e-5,
  "epochs": 3,
  "device": "cuda"
}