{ "_name_or_path": "microsoft/swinv2-tiny-patch4-window8-256", "architectures": [ "Swinv2ForImageClassification" ], "attention_probs_dropout_prob": 0.0, "depths": [ 2, 2, 6, 2 ], "drop_path_rate": 0.1, "embed_dim": 96, "encoder_stride": 32, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "11214 Bush 3M friction with Cross axle", "1": "18651 Cross Axle 2M with Snap friction", "2": "2357 Brick corner 1x2x2", "3": "3003 Brick 2x2", "4": "3004 Brick 1x2", "5": "3005 Brick 1x1", "6": "3022 Plate 2x2", "7": "3023 Plate 1x2", "8": "3024 Plate 1x1", "9": "3040 Roof Tile 1x2x45deg", "10": "3069 Flat Tile 1x2", "11": "32123 half Bush", "12": "3673 Peg 2M", "13": "3713 Bush for Cross Axle", "14": "3794 Plate 1X2 with 1 Knob", "15": "6632 Technic Lever 3M" }, "image_size": 256, "initializer_range": 0.02, "label2id": { "11214 Bush 3M friction with Cross axle": 0, "18651 Cross Axle 2M with Snap friction": 1, "2357 Brick corner 1x2x2": 2, "3003 Brick 2x2": 3, "3004 Brick 1x2": 4, "3005 Brick 1x1": 5, "3022 Plate 2x2": 6, "3023 Plate 1x2": 7, "3024 Plate 1x1": 8, "3040 Roof Tile 1x2x45deg": 9, "3069 Flat Tile 1x2": 10, "32123 half Bush": 11, "3673 Peg 2M": 12, "3713 Bush for Cross Axle": 13, "3794 Plate 1X2 with 1 Knob": 14, "6632 Technic Lever 3M": 15 }, "layer_norm_eps": 1e-05, "mlp_ratio": 4.0, "model_type": "swinv2", "num_channels": 3, "num_heads": [ 3, 6, 12, 24 ], "num_layers": 4, "out_features": [ "stage4" ], "out_indices": [ 4 ], "patch_size": 4, "path_norm": true, "pretrained_window_sizes": [ 0, 0, 0, 0 ], "problem_type": "single_label_classification", "qkv_bias": true, "stage_names": [ "stem", "stage1", "stage2", "stage3", "stage4" ], "torch_dtype": "float32", "transformers_version": "4.40.2", "use_absolute_embeddings": false, "window_size": 8 }