split50/config.json

{
  "add_feats": false,
  "architectures": [
    "Network"
  ],
  "base_channels": 32,
  "channels_multiplier": 1.8,
  "expansion_rate": 2.1,
  "in_channels": 1,
  "n_blocks": [
    3,
    2,
    1
  ],
  "n_classes": 10,
  "strides": {
    "b2": [
      1,
      1
    ],
    "b3": [
      1,
      2
    ],
    "b4": [
      2,
      1
    ]
  },
  "torch_dtype": "float16",
  "transformers_version": "4.37.1"
}