hanslab37's picture
Training in progress, epoch 1
097fe16 verified
{
"_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": "Achaemenid architecture",
"1": "American Foursquare architecture",
"10": "Chicago school architecture",
"11": "Colonial architecture",
"12": "Deconstructivism",
"13": "Edwardian architecture",
"14": "Georgian architecture",
"15": "Gothic architecture",
"16": "Greek Revival architecture",
"17": "International style",
"18": "Novelty architecture",
"19": "Palladian architecture",
"2": "American craftsman style",
"20": "Postmodern architecture",
"21": "Queen Anne architecture",
"22": "Romanesque architecture",
"23": "Russian Revival architecture",
"24": "Tudor Revival architecture",
"3": "Ancient Egyptian architecture",
"4": "Art Deco architecture",
"5": "Art Nouveau architecture",
"6": "Baroque architecture",
"7": "Bauhaus architecture",
"8": "Beaux-Arts architecture",
"9": "Byzantine architecture"
},
"label2id": {
"Achaemenid architecture": "0",
"American Foursquare architecture": "1",
"American craftsman style": "2",
"Ancient Egyptian architecture": "3",
"Art Deco architecture": "4",
"Art Nouveau architecture": "5",
"Baroque architecture": "6",
"Bauhaus architecture": "7",
"Beaux-Arts architecture": "8",
"Byzantine architecture": "9",
"Chicago school architecture": "10",
"Colonial architecture": "11",
"Deconstructivism": "12",
"Edwardian architecture": "13",
"Georgian architecture": "14",
"Gothic architecture": "15",
"Greek Revival architecture": "16",
"International style": "17",
"Novelty architecture": "18",
"Palladian architecture": "19",
"Postmodern architecture": "20",
"Queen Anne architecture": "21",
"Romanesque architecture": "22",
"Russian Revival architecture": "23",
"Tudor Revival architecture": "24"
},
"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.41.1"
}