RuntimeError: Cannot find callable deit_tiny_patch16_224 in hubconf

1

Hi,

Could you someone help me understand why I got this error “RuntimeError: Cannot find callable deit_tiny_patch16_224 in hubconf”, when running the following code:

    model = torch.hub.load('facebookresearch/deit:main',
                                   'deit_tiny_patch16_224', pretrained=True)
2

I cannot reproduce the issue using a recent nightly build:

model = torch.hub.load('facebookresearch/deit:main',
                       'deit_tiny_patch16_224', pretrained=True)

model
# VisionTransformer(
#   (patch_embed): PatchEmbed(
#     (proj): Conv2d(3, 192, kernel_size=(16, 16), stride=(16, 16))
#     (norm): Identity()
#   )
#   (pos_drop): Dropout(p=0.0, inplace=False)
#   (patch_drop): Identity()
#   (norm_pre): Identity()
#   (blocks): Sequential(
#     (0): Block(
# ...
1 Like
3

Thanks for your reply. I have fixed it. It was caused by another custom module.