config.json

{
  "architectures": [
    "fcnet"
  ],
  "conv1": [
    1,
    32,
    3,
    1
  ],
  "conv2": [
    32,
    64,
    3,
    1
  ],
  "dropout1": [
    0.25
  ],
  "dropout2": [
    0.5
  ],
  "fc1": [
    9216,
    128
  ],
  "fc2": [
    128,
    10
  ],
  "model_type": "fcnet",
  "torch_dtype": "float32",
  "transformers_version": "4.18.0.release"
}