我嘗試訓練我的神經網絡,然后評估它的測試準確性。我正在使用本文底部的代碼進行訓練。事實上,對于其他神經網絡,我可以毫無問題地使用我的代碼評估測試準確性。然而,對于這個神經網絡(我根據神經網絡論文的描述正確構建),我無法正確評估測試精度,它給我下面的回溯。所以也許我的前傳有問題?模型代碼在這里,包括前向傳遞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 回答

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