{ "_name_or_path": "google/vit-base-patch16-224", "architectures": [ "ViTForImageClassification" ], "attention_probs_dropout_prob": 0.0, "encoder_stride": 16, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "Atelectasis", "1": "Cardiomegaly", "2": "Consolidation", "3": "Edema", "4": "Effusion", "5": "Emphysema", "6": "Fibrosis", "7": "Hernia", "8": "Infiltration", "9": "Mass", "10": "No Finding", "11": "Nodule", "12": "Pleural_Thickening", "13": "Pneumonia", "14": "Pneumothorax" }, "image_size": 224, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "Atelectasis": 0, "Cardiomegaly": 1, "Consolidation": 2, "Edema": 3, "Effusion": 4, "Emphysema": 5, "Fibrosis": 6, "Hernia": 7, "Infiltration": 8, "Mass": 9, "No Finding": 10, "Nodule": 11, "Pleural_Thickening": 12, "Pneumonia": 13, "Pneumothorax": 14 }, "layer_norm_eps": 1e-12, "model_type": "vit", "num_attention_heads": 12, "num_channels": 3, "num_hidden_layers": 12, "patch_size": 16, "problem_type": "single_label_classification", "qkv_bias": true, "torch_dtype": "float32", "transformers_version": "4.36.0.dev0" }