facebookresearch / deit

Official DeiT repository
Apache License 2.0
3.94k stars 547 forks source link

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

Open GerogeD opened 10 months ago

GerogeD commented 10 months ago

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~