There is no pretrained model for vit_b_16
Traceback (most recent call last):
File "train.py", line 225, in <module>
model = get_model_from_name[backbone](input_shape = input_shape, num_classes = num_classes, pretrained = pretrained)
File "/data/xly/hhs/classification-pytorch-1/nets/vision_transformer.py", line 223, in vit_b_16
model.load_state_dict(torch.load("/data/xly/hhs/classification-pytorch-1/model_data/vit-patch_16.pth"))
File "/root/anaconda3/envs/torch38/lib/python3.8/site-packages/torch/serialization.py", line 595, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/root/anaconda3/envs/torch38/lib/python3.8/site-packages/torch/serialization.py", line 749, in _legacy_load
return legacy_load(f)
File "/root/anaconda3/envs/torch38/lib/python3.8/site-packages/torch/serialization.py", line 674, in legacy_load
tar.extract('storages', path=tmpdir)
File "/root/anaconda3/envs/torch38/lib/python3.8/tarfile.py", line 2272, in extract
tarinfo = self._get_extract_tarinfo(member, filter_function, path)
File "/root/anaconda3/envs/torch38/lib/python3.8/tarfile.py", line 2279, in _get_extract_tarinfo
tarinfo = self.getmember(member)
File "/root/anaconda3/envs/torch38/lib/python3.8/tarfile.py", line 1962, in getmember
raise KeyError("filename %r not found" % name)
KeyError: "filename 'storages' not found"