Vit预训练权重

[ScottXiao233]

出错信息：

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"

权重文件： 环境： Ubuntu 20.04 其余配置和项目中的requirements一样 就是权重文件损坏了吗？