IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
37141d4 verified
Raw
History Blame Contribute Delete
609 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.swin",
"model_class": "SwinImageClassify",
"variant": "swin_tiny_patch4_window7_224_ms_in1k",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "swin",
"vision_config": {
"window_size": 7,
"embed_dim": 96,
"depths": [
2,
2,
6,
2
],
"num_heads": [
3,
6,
12,
24
],
"pretrain_size": 224,
"image_size": 224,
"num_classes": 1000
}
}