{ "_name_or_path": "microsoft/resnet-18", "architectures": [ "CifNetForImageClassification" ], "embedding_kwargs": { "__comment": "First hidden layer size", "embedding_size": 32, "embedding_kernel_size_1": 7, "embedding_kernel_size_2": 2, "embedding_stride_1": 2, "embedding_stride_2": 2 }, "depths": [ 1, 2, 4, 3 ], "activation": "silu", "hidden_sizes": [ 64, 64, 128, 256 ], "id2label": { "0": "airplane", "1": "automobile", "2": "bird", "3": "cat", "4": "deer", "5": "dog", "6": "frog", "7": "horse", "8": "ship", "9": "truck" }, "label2id": { "airplane": "0", "automobile": "1", "bird": "2", "cat": "3", "deer": "4", "dog": "5", "frog": "6", "horse": "7", "ship": "8", "truck": "9" }, "layer_type": "basic", "model_type": "resnet", "num_channels": 3, "torch_dtype": "float32", "transformers_version": "4.38.2" }