config.json

{
  "model_type": "digit_recognition",
  "architectures": [
    "DigitRecognitionModel"
  ],
  "attention_heads": 1,
  "hidden_size": 32,
  "intermediate_size": 128,
  "num_labels": 10,
  "vocab_size": 784
}