{ | |
"_name_or_path": "facebook/levit-192", | |
"architectures": [ | |
"LevitForImageClassification" | |
], | |
"attention_ratio": [ | |
2, | |
2, | |
2 | |
], | |
"depths": [ | |
4, | |
4, | |
4 | |
], | |
"down_ops": [ | |
[ | |
"Subsample", | |
32, | |
6, | |
4, | |
2, | |
2 | |
], | |
[ | |
"Subsample", | |
32, | |
9, | |
4, | |
2, | |
2 | |
], | |
[ | |
"" | |
] | |
], | |
"drop_path_rate": 0, | |
"hidden_sizes": [ | |
192, | |
288, | |
384 | |
], | |
"id2label": { | |
"0": "illustrated", | |
"1": "not-illustrated" | |
}, | |
"image_size": 224, | |
"initializer_range": 0.02, | |
"kernel_size": 3, | |
"key_dim": [ | |
32, | |
32, | |
32 | |
], | |
"label2id": { | |
"illustrated": 0, | |
"not-illustrated": 1 | |
}, | |
"mlp_ratio": [ | |
2, | |
2, | |
2 | |
], | |
"model_type": "levit", | |
"num_attention_heads": [ | |
3, | |
5, | |
6 | |
], | |
"num_channels": 3, | |
"padding": 1, | |
"patch_size": 16, | |
"problem_type": "single_label_classification", | |
"stride": 2, | |
"torch_dtype": "float32", | |
"transformers_version": "4.22.2" | |
} | |