IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
aeb6d80 verified
Raw
History Blame Contribute Delete
579 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.deit",
"model_class": "DeiTImageClassify",
"variant": "deit3_large_patch16_384_fb_in1k",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "deit",
"vision_config": {
"patch_size": 16,
"embed_dim": 1024,
"depth": 24,
"num_heads": 16,
"use_distillation": false,
"no_embed_class": true,
"layer_scale_init": 1e-06,
"image_size": 384,
"num_classes": 1000
}
}