亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

計算神經網絡測試精度的錯誤

計算神經網絡測試精度的錯誤

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

您需要使用之前保存過狀態指令的同一模型。


查看完整回答
反對 回復 2023-01-04
  • 1 回答
  • 0 關注
  • 121 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號