{ "_name_or_path": "facebook/levit-128", "architectures": [ "LevitForImageClassification" ], "attention_ratio": [ 2, 2, 2 ], "depths": [ 4, 4, 4 ], "down_ops": [ [ "Subsample", 16, 8, 4, 2, 2 ], [ "Subsample", 16, 16, 4, 2, 2 ], [ "" ] ], "drop_path_rate": 0, "hidden_sizes": [ 128, 256, 384 ], "id2label": { "0": "astilbe", "1": "bellflower", "2": "black_eyed_susan", "3": "calendula", "4": "california_poppy", "5": "carnation", "6": "common_daisy", "7": "coreopsis", "8": "daffodil", "9": "dandelion", "10": "iris", "11": "magnolia", "12": "rose", "13": "sunflower", "14": "tulip", "15": "water_lily" }, "image_size": 224, "initializer_range": 0.02, "kernel_size": 3, "key_dim": [ 16, 16, 16 ], "label2id": { "astilbe": 0, "bellflower": 1, "black_eyed_susan": 2, "calendula": 3, "california_poppy": 4, "carnation": 5, "common_daisy": 6, "coreopsis": 7, "daffodil": 8, "dandelion": 9, "iris": 10, "magnolia": 11, "rose": 12, "sunflower": 13, "tulip": 14, "water_lily": 15 }, "mlp_ratio": [ 2, 2, 2 ], "model_type": "levit", "num_attention_heads": [ 4, 8, 12 ], "num_channels": 3, "padding": 1, "patch_size": 16, "problem_type": "single_label_classification", "stride": 2, "torch_dtype": "float32", "transformers_version": "4.39.3" }