config.json

{
  "model_type": "NULA",
  "num_classes": 10,
  "in_channels": 3,
  "input_size": [3, 32, 32],
  "block_channels": [64, 128, 256],
  "use_residual": true,
  "use_se": true,
  "use_spatial_attention": false,
  "norm_layer": "batchnorm",
  "activation": "relu",
  "classifier_hidden_dim": 256,
  "init_scheme": "kaiming_normal",
  "dataset": "uoft-cs/cifar10",
  "mean": [0.5, 0.5, 0.5],
  "std": [0.5, 0.5, 0.5],
  "id2label": {
    "0": "airplane",
    "1": "automobile",
    "2": "bird",
    "3": "cat",
    "4": "deer",
    "5": "dog",
    "6": "frog",
    "7": "horse",
    "8": "ship",
    "9": "truck"
  },
  "label2id": {
    "airplane": 0,
    "automobile": 1,
    "bird": 2,
    "cat": 3,
    "deer": 4,
    "dog": 5,
    "frog": 6,
    "horse": 7,
    "ship": 8,
    "truck:" 9
  }
}