| { | |
| "model_type": "image_classification", | |
| "architecture": "ViT", | |
| "pretrained_model_name": "google/vit-base-patch16-224", | |
| "num_classes": 2, | |
| "input_size": [224, 224], | |
| "input_channels": 3, | |
| "output_size": 2, | |
| "mean": [0.5, 0.5, 0.5], | |
| "std": [0.5, 0.5, 0.5], | |
| "max_seq_length": null, | |
| "pad_to_max_seq_length": false, | |
| "truncation_strategy": "longest_first", | |
| "do_lower_case": false, | |
| "tokenizer_name": null, | |
| "transformer_model_type": null, | |
| "config_name": null, | |
| "dropout": 0.1, | |
| "attention_dropout": 0.1, | |
| "hidden_dropout": 0.1, | |
| "num_attention_heads": 12, | |
| "hidden_size": 768, | |
| "intermediate_size": 3072, | |
| "num_hidden_layers": 12, | |
| "activation_function": "gelu", | |
| "initializer_range": 0.02, | |
| "layer_norm_eps": 1e-12, | |
| "gradient_checkpointing": false, | |
| "use_cache": true, | |
| "output_attentions": false, | |
| "output_hidden_states": false, | |
| "id2label": {"0": "not hotdog", "1": "hotdog"}, | |
| "label2id": {"not hotdog": 0, "hotdog": 1} | |
| } | |