{
  "hidden_dims": 128,
  "num_layers": 4,
  "num_heads": 8,
  "dropout_rate": 0.2,
  "batch_size": 128,
  "learning_rate": "1e-4",
  "weight_decay": "1e-4"
}