Card: load with fgir_zoo create_model

README.md

@@ -73,7 +73,7 @@ from PIL import Image
 from torchvision import transforms
 from fgir_zoo import backbones
 
-model = backbones.create_backbone('vit_b16', dataset='cub', strategy='cal')  # cal_224/cub_vit_b16_cal.pth
+model = backbones.create_model('cal_224/cub_vit_b16_cal')  # or model_name='vit_b16', dataset='cub', strategy='cal'
 cfg = model.config
 tf = transforms.Compose([
     transforms.Resize((cfg.test_resize_size, cfg.test_resize_size),