{
  "_name_or_path": "google/efficientnet-b0",
  "architectures": [
    "EfficientNetForImageClassification"
  ],
  "batch_norm_eps": 0.001,
  "batch_norm_momentum": 0.99,
  "depth_coefficient": 1.0,
  "depth_divisor": 8,
  "depthwise_padding": [],
  "drop_connect_rate": 0.2,
  "dropout_rate": 0.2,
  "expand_ratios": [
    1,
    6,
    6,
    6,
    6,
    6,
    6
  ],
  "hidden_act": "swish",
  "hidden_dim": 1280,
  "id2label": {
    "0": "AL2",
    "1": "AL4",
    "2": "AL6",
    "3": "AS4",
    "4": "AS6",
    "5": "BB2",
    "6": "BB3",
    "7": "BB4",
    "8": "BB5",
    "9": "Bent",
    "10": "D",
    "11": "D_dirt",
    "12": "Dirty",
    "13": "GG",
    "14": "H_015_31",
    "15": "H_020_31",
    "16": "H_025_31",
    "17": "H_030_31",
    "18": "H_035_31",
    "19": "H_040_31",
    "20": "H_045_31",
    "21": "H_050_31",
    "22": "H_055_31",
    "23": "H_060_31",
    "24": "H_070_31",
    "25": "H_080_31",
    "26": "H_090_31",
    "27": "H_100_31",
    "28": "H_110_31",
    "29": "H_120_31",
    "30": "K_008_25",
    "31": "K_010_21",
    "32": "K_010_25",
    "33": "K_010_31",
    "34": "K_015_21",
    "35": "K_015_25",
    "36": "K_015_31",
    "37": "K_020_21",
    "38": "K_020_25",
    "39": "K_020_31",
    "40": "K_025_21",
    "41": "K_025_25",
    "42": "K_025_31",
    "43": "K_030_21",
    "44": "K_030_25",
    "45": "K_030_31",
    "46": "K_035_21",
    "47": "K_035_25",
    "48": "K_035_31",
    "49": "K_040_21",
    "50": "K_040_25",
    "51": "K_040_31",
    "52": "K_045_21",
    "53": "K_045_25",
    "54": "K_045_31",
    "55": "K_050_21",
    "56": "K_050_25",
    "57": "K_050_31",
    "58": "K_055_21",
    "59": "K_055_25",
    "60": "K_055_31",
    "61": "K_060_21",
    "62": "K_060_25",
    "63": "K_060_31",
    "64": "K_070_21",
    "65": "K_070_25",
    "66": "K_070_31",
    "67": "K_080_21",
    "68": "K_080_25",
    "69": "K_080_31",
    "70": "K_090_25",
    "71": "K_090_31",
    "72": "K_100_25",
    "73": "K_100_31",
    "74": "K_110_25",
    "75": "K_110_31",
    "76": "K_120_25",
    "77": "K_130_25",
    "78": "K_140_25",
    "79": "PC_Black",
    "80": "PC_Blue",
    "81": "PC_Green",
    "82": "RL10",
    "83": "RL14",
    "84": "RL18",
    "85": "RS5",
    "86": "RS7"
  },
  "image_size": 224,
  "in_channels": [
    32,
    16,
    24,
    40,
    80,
    112,
    192
  ],
  "initializer_range": 0.02,
  "kernel_sizes": [
    3,
    3,
    5,
    3,
    5,
    5,
    3
  ],
  "label2id": {
    "AL2": 0,
    "AL4": 1,
    "AL6": 2,
    "AS4": 3,
    "AS6": 4,
    "BB2": 5,
    "BB3": 6,
    "BB4": 7,
    "BB5": 8,
    "Bent": 9,
    "D": 10,
    "D_dirt": 11,
    "Dirty": 12,
    "GG": 13,
    "H_015_31": 14,
    "H_020_31": 15,
    "H_025_31": 16,
    "H_030_31": 17,
    "H_035_31": 18,
    "H_040_31": 19,
    "H_045_31": 20,
    "H_050_31": 21,
    "H_055_31": 22,
    "H_060_31": 23,
    "H_070_31": 24,
    "H_080_31": 25,
    "H_090_31": 26,
    "H_100_31": 27,
    "H_110_31": 28,
    "H_120_31": 29,
    "K_008_25": 30,
    "K_010_21": 31,
    "K_010_25": 32,
    "K_010_31": 33,
    "K_015_21": 34,
    "K_015_25": 35,
    "K_015_31": 36,
    "K_020_21": 37,
    "K_020_25": 38,
    "K_020_31": 39,
    "K_025_21": 40,
    "K_025_25": 41,
    "K_025_31": 42,
    "K_030_21": 43,
    "K_030_25": 44,
    "K_030_31": 45,
    "K_035_21": 46,
    "K_035_25": 47,
    "K_035_31": 48,
    "K_040_21": 49,
    "K_040_25": 50,
    "K_040_31": 51,
    "K_045_21": 52,
    "K_045_25": 53,
    "K_045_31": 54,
    "K_050_21": 55,
    "K_050_25": 56,
    "K_050_31": 57,
    "K_055_21": 58,
    "K_055_25": 59,
    "K_055_31": 60,
    "K_060_21": 61,
    "K_060_25": 62,
    "K_060_31": 63,
    "K_070_21": 64,
    "K_070_25": 65,
    "K_070_31": 66,
    "K_080_21": 67,
    "K_080_25": 68,
    "K_080_31": 69,
    "K_090_25": 70,
    "K_090_31": 71,
    "K_100_25": 72,
    "K_100_31": 73,
    "K_110_25": 74,
    "K_110_31": 75,
    "K_120_25": 76,
    "K_130_25": 77,
    "K_140_25": 78,
    "PC_Black": 79,
    "PC_Blue": 80,
    "PC_Green": 81,
    "RL10": 82,
    "RL14": 83,
    "RL18": 84,
    "RS5": 85,
    "RS7": 86
  },
  "model_type": "efficientnet",
  "num_block_repeats": [
    1,
    2,
    2,
    3,
    3,
    4,
    1
  ],
  "num_channels": 3,
  "num_hidden_layers": 64,
  "out_channels": [
    16,
    24,
    40,
    80,
    112,
    192,
    320
  ],
  "out_features": null,
  "pooling_type": "mean",
  "problem_type": "multi_label_classification",
  "squeeze_expansion_ratio": 0.25,
  "stage_names": [
    "stem",
    "stage1",
    "stage2",
    "stage3",
    "stage4",
    "stage5",
    "stage6",
    "stage7"
  ],
  "strides": [
    1,
    2,
    2,
    2,
    1,
    2,
    1
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.35.2",
  "width_coefficient": 1.0
}