{ "_name_or_path": "google/vit-base-patch16-224-in21k", "architectures": [ "ViTForImageClassification" ], "attention_probs_dropout_prob": 0.0, "encoder_stride": 16, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "No Finding", "1": "Atelectasis", "2": "Cardiomegaly", "3": "Effusion", "4": "Infiltration", "5": "Mass", "6": "Nodule", "7": "Pneumonia", "8": "Pneumothorax", "9": "Consolidation", "10": "Edema", "11": "Emphysema", "12": "Fibrosis", "13": "Pleural_Thickening", "14": "Hernia" }, "image_size": 224, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "Atelectasis": 1, "Cardiomegaly": 2, "Consolidation": 9, "Edema": 10, "Effusion": 3, "Emphysema": 11, "Fibrosis": 12, "Hernia": 14, "Infiltration": 4, "Mass": 5, "No Finding": 0, "Nodule": 6, "Pleural_Thickening": 13, "Pneumonia": 7, "Pneumothorax": 8 }, "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.34.1" }