Team Ai
Modelpublic

cppgohan/my_model

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes27downloads
config.json50 linesDownload Raw Back to root
1{2  "_name_or_path": "microsoft/resnet-50",3  "architectures": [4    "ResNetForImageClassification"5  ],6  "depths": [7    3,8    4,9    6,10    311  ],12  "downsample_in_bottleneck": false,13  "downsample_in_first_stage": false,14  "embedding_size": 64,15  "hidden_act": "relu",16  "hidden_sizes": [17    256,18    512,19    1024,20    204821  ],22  "id2label": {23    "0": "cat",24    "1": "dog"25  },26  "label2id": {27    "cat": "0",28    "dog": "1"29  },30  "layer_type": "bottleneck",31  "model_type": "resnet",32  "num_channels": 3,33  "out_features": [34    "stage4"35  ],36  "out_indices": [37    438  ],39  "problem_type": "single_label_classification",40  "stage_names": [41    "stem",42    "stage1",43    "stage2",44    "stage3",45    "stage4"46  ],47  "torch_dtype": "float32",48  "transformers_version": "4.38.2"49}50