config.json

{
  "model_type": "BaseCNN",
  "model_config": {
    "layers": [
      32,
      64,
      128,
      256
    ],
    "kernel": 3,
    "padding": 1,
    "stride": 1,
    "batchnorm": true,
    "bn_pre_activ": true,
    "activation": "ReLU",
    "dropout": 0.4,
    "pool": true,
    "fc_layers": [
      256,
      128
    ],
    "nr_classes": 3,
    "in_channels": 3
  },
  "num_labels": 3,
  "id2label": {
    "0": "Type 1",
    "1": "Type 2",
    "2": "Type 3"
  },
  "label2id": {
    "Type 1": 0,
    "Type 2": 1,
    "Type 3": 2
  }
}