| { | |
| "_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 | |
| } | |