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
Hello, when i want to use "deit_tiny_distilled_patch16_224" but i can't find it.
Transform
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~