Skip to content
This repository has been archived by the owner on Mar 15, 2024. It is now read-only.

Hi,Why can't I find deit_tiny_distilled_patch16_224 in hubconf #231

Open
GerogeD opened this issue Aug 14, 2023 · 0 comments
Open

Hi,Why can't I find deit_tiny_distilled_patch16_224 in hubconf #231

GerogeD opened this issue Aug 14, 2023 · 0 comments

Comments

@GerogeD
Copy link

GerogeD commented Aug 14, 2023

Hello, when i want to use "deit_tiny_distilled_patch16_224" but i can't find it.

Transform

    transformer = torch.hub.load('facebookresearch/deit:main', 'deit_tiny_distilled_patch16_224', pretrained=True) # tiny_distilled
    self.patch_embed = transformer.patch_embed
    self.transformers = nn.ModuleList(
        [transformer.blocks[i] for i in range(12)]
    )
    
  Traceback (most recent call last):

File "/home/common-dir/HX_code/PANet/Pri_train.py", line 429, in
main()
File "/home/common-dir/HX_code/PANet/Pri_train.py", line 128, in main
leaky=use_leaky, norm=norm_type).to(device)
File "/home/common-dir/HX_code/PANet/models/networks.py", line 274, in init
transformer = torch.hub.load('facebookresearch/deit:main', 'deit_tiny_distilled_patch16_224', pretrained=True) # tiny_distilled
File "/root/miniconda3/envs/py37/lib/python3.7/site-packages/torch/hub.py", line 370, in load
model = _load_local(repo_or_dir, model, *args, **kwargs)
File "/root/miniconda3/envs/py37/lib/python3.7/site-packages/torch/hub.py", line 398, in _load_local
entry = _load_entry_from_hubconf(hub_module, model)
File "/root/miniconda3/envs/py37/lib/python3.7/site-packages/torch/hub.py", line 218, in _load_entry_from_hubconf
raise RuntimeError('Cannot find callable {} in hubconf'.format(model))
RuntimeError: Cannot find callable deit_tiny_distilled_patch16_224 in hubconf

Thank you very much if you can help me~

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant