{ "architectures": [ "BitForImageClassification" ], "depths": [ 1, 1, 2, 1 ], "drop_path_rate": 0.0, "embedding_dynamic_padding": false, "embedding_size": 64, "embeddings_size": 10, "global_padding": null, "hidden_act": "relu", "hidden_sizes": [ 8, 16, 32, 64 ], "id2label": { "0": "LABEL_0", "1": "LABEL_1", "2": "LABEL_2" }, "label2id": { "LABEL_0": 0, "LABEL_1": 1, "LABEL_2": 2 }, "layer_type": "preactivation", "model_type": "bit", "num_channels": 3, "num_groups": 1, "out_features": [ "stage2", "stage3", "stage4" ], "output_stride": 32, "stage_names": [ "stem", "stage1", "stage2", "stage3", "stage4" ], "torch_dtype": "float32", "transformers_version": "4.28.0.dev0", "width_factor": 1 }