config.json

{
  "attn_dropout": 0.0,
  "dim": 512,
  "n_classes": 10,
  "n_features": 100,
  "n_heads": 4,
  "n_layers": 12,
  "task": "regression",
  "y_as_float_embedding": true
}