ruoruoruo666的博客

RuntimeError(‘Error(s) in loading state_dict for {}:\n\t{}’.format将原来代码model.load_state_dict(torch.load(dir))改为model.load_state_dict(torch.load(dir), False)

将预训练好的模型加载到新的网络上。import torchfrom vit_pytorch import ViTmodel = ViT( image_size = 256, patch_size = 8, num_classes = 33, dim = 256, depth = 6, heads = 4, mlp_dim = 256, dropout = 0.1, emb_dropout = 0.1)#params=model

train.pyimport torchfrom torch import nnfrom torch.nn import functional as Ffrom torchvision import transformsfrom torch.utils.data import DataLoader,Datasetfrom torch import optimimport osimport csvfrom PIL import Imageimport warningswarnings.s