我嘗試訓(xùn)練我的神經(jīng)網(wǎng)絡(luò),然后評估它的測試準(zhǔn)確性。我正在使用本文底部的代碼進行訓(xùn)練。事實上,對于其他神經(jīng)網(wǎng)絡(luò),我可以毫無問題地使用我的代碼評估測試準(zhǔn)確性。然而,對于這個神經(jīng)網(wǎng)絡(luò)(我根據(jù)神經(jīng)網(wǎng)絡(luò)論文的描述正確構(gòu)建),我無法正確評估測試精度,它給我下面的回溯。所以也許我的前傳有問題?模型代碼在這里,包括前向傳遞import torchimport torch.nn as nnF = nn.functional__all__ = ['cifar10_deepnet', 'cifar100_deepnet']class VGG(nn.Module): def __init__(self, num_classes=10): super(VGG, self).__init__() self.features = nn.Sequential( nn.Conv2d(3, 64, kernel_size=3, bias=False), nn.BatchNorm2d(64), nn.ReLU(inplace=True), nn.Dropout(0.3), nn.Conv2d(64, 64, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(64), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2, stride=2), nn.Conv2d(64, 128, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(128), nn.ReLU(inplace=True), nn.Dropout(0.4), nn.Conv2d(128, 128, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(128), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2, stride=2), nn.Conv2d(128, 256, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(256), nn.ReLU(inplace=True), nn.Dropout(0.4), nn.Conv2d(256, 256, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(256), nn.ReLU(inplace=True), nn.Dropout(0.4), nn.Conv2d(256, 256, kernel_size=3, padding = 1, bias=False), nn.BatchNorm2d(256), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2, stride=2),
1 回答

當(dāng)年話下
TA貢獻1890條經(jīng)驗 獲得超9個贊
您正在嘗試加載屬于另一個模型的狀態(tài)字典。
該錯誤表明您的模型是 class AlexNet
。
RunTimeError: Error(s) in loading state_dict for AlexNet:
但是您嘗試加載的狀態(tài)字典來自VGG
您發(fā)布的內(nèi)容,它與AlexNet
.
您需要使用之前保存過狀態(tài)指令的同一模型。
添加回答
舉報
0/150
提交
取消