oxford_flowers_model / config.json
paolosoprani's picture
Training in progress, epoch 0
e1e6304 verified
raw
history blame
3.96 kB
{
"_name_or_path": "microsoft/resnet-50",
"architectures": [
"ResNetForImageClassification"
],
"depths": [
3,
4,
6,
3
],
"downsample_in_bottleneck": false,
"downsample_in_first_stage": false,
"embedding_size": 64,
"hidden_act": "relu",
"hidden_sizes": [
256,
512,
1024,
2048
],
"id2label": {
"0": "1",
"1": "10",
"10": "16",
"100": "98",
"101": "99",
"11": "17",
"12": "18",
"13": "19",
"14": "2",
"15": "20",
"16": "21",
"17": "22",
"18": "23",
"19": "24",
"2": "100",
"20": "25",
"21": "26",
"22": "27",
"23": "28",
"24": "29",
"25": "3",
"26": "30",
"27": "31",
"28": "32",
"29": "33",
"3": "101",
"30": "34",
"31": "35",
"32": "36",
"33": "37",
"34": "38",
"35": "39",
"36": "4",
"37": "40",
"38": "41",
"39": "42",
"4": "102",
"40": "43",
"41": "44",
"42": "45",
"43": "46",
"44": "47",
"45": "48",
"46": "49",
"47": "5",
"48": "50",
"49": "51",
"5": "11",
"50": "52",
"51": "53",
"52": "54",
"53": "55",
"54": "56",
"55": "57",
"56": "58",
"57": "59",
"58": "6",
"59": "60",
"6": "12",
"60": "61",
"61": "62",
"62": "63",
"63": "64",
"64": "65",
"65": "66",
"66": "67",
"67": "68",
"68": "69",
"69": "7",
"7": "13",
"70": "70",
"71": "71",
"72": "72",
"73": "73",
"74": "74",
"75": "75",
"76": "76",
"77": "77",
"78": "78",
"79": "79",
"8": "14",
"80": "8",
"81": "80",
"82": "81",
"83": "82",
"84": "83",
"85": "84",
"86": "85",
"87": "86",
"88": "87",
"89": "88",
"9": "15",
"90": "89",
"91": "9",
"92": "90",
"93": "91",
"94": "92",
"95": "93",
"96": "94",
"97": "95",
"98": "96",
"99": "97"
},
"label2id": {
"1": "0",
"10": "1",
"100": "2",
"101": "3",
"102": "4",
"11": "5",
"12": "6",
"13": "7",
"14": "8",
"15": "9",
"16": "10",
"17": "11",
"18": "12",
"19": "13",
"2": "14",
"20": "15",
"21": "16",
"22": "17",
"23": "18",
"24": "19",
"25": "20",
"26": "21",
"27": "22",
"28": "23",
"29": "24",
"3": "25",
"30": "26",
"31": "27",
"32": "28",
"33": "29",
"34": "30",
"35": "31",
"36": "32",
"37": "33",
"38": "34",
"39": "35",
"4": "36",
"40": "37",
"41": "38",
"42": "39",
"43": "40",
"44": "41",
"45": "42",
"46": "43",
"47": "44",
"48": "45",
"49": "46",
"5": "47",
"50": "48",
"51": "49",
"52": "50",
"53": "51",
"54": "52",
"55": "53",
"56": "54",
"57": "55",
"58": "56",
"59": "57",
"6": "58",
"60": "59",
"61": "60",
"62": "61",
"63": "62",
"64": "63",
"65": "64",
"66": "65",
"67": "66",
"68": "67",
"69": "68",
"7": "69",
"70": "70",
"71": "71",
"72": "72",
"73": "73",
"74": "74",
"75": "75",
"76": "76",
"77": "77",
"78": "78",
"79": "79",
"8": "80",
"80": "81",
"81": "82",
"82": "83",
"83": "84",
"84": "85",
"85": "86",
"86": "87",
"87": "88",
"88": "89",
"89": "90",
"9": "91",
"90": "92",
"91": "93",
"92": "94",
"93": "95",
"94": "96",
"95": "97",
"96": "98",
"97": "99",
"98": "100",
"99": "101"
},
"layer_type": "bottleneck",
"model_type": "resnet",
"num_channels": 3,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"problem_type": "single_label_classification",
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.39.3"
}