config.json

{
  "model_type": "custom_model",
  "architectures": [
    "Model"
  ],
  "projection_dim": 128,
  "num_classes": 4
}