將2個(gè)項(xiàng)目合并的時(shí)候，把文件夾A塞到文件夾B中，作為一個(gè)子文件夾，文件夾A中有一個(gè)已經(jīng)訓(xùn)練好的.pth模型文件，B調(diào)用A以加載這個(gè)模型的時(shí)候，出現(xiàn)了題目中的報(bào)錯(cuò)，具體情況如下圖

image.png

時(shí)間久遠(yuǎn)，忘記這個(gè).pth模型是怎么save的了，但是查了資料發(fā)現(xiàn)可能是因?yàn)檫@個(gè)模型保存了除模型參數(shù)之外其他的參數(shù)，因?yàn)樯婕暗轿募木唧w位置，所以在將A塞到B中的時(shí)候就會(huì)發(fā)生錯(cuò)誤，由于舊的.pth保留的好，就將其重新load然后僅取其中的模型參數(shù)再save成新的.pth模型

···

重新加載和save代碼

package = torch.load(model_path)
p = './static_dict.pth'
torch.save(package.state_dict(), p)
···

···

調(diào)用新的模型代碼

model = Model_CNN()
model.load_state_dict(torch.load(model_path))
model.eval()
model = model.to(device)
···