1 回答

TA貢獻1880條經驗 獲得超4個贊
你的理解是正確的,非常詳細。
但是,您使用了兩個池化層(請參閱下面的相關代碼)。所以第二步之后的輸出將是16個44/2=22維度的地圖。
x = self.pool(F.relu(self.conv1(x)))
x = self.pool(F.relu(self.conv2(x)))
要解決此問題,要么不池化,要么將全連接層的維度更改為22*22*16。
要通過不池化來修復,請修改您的轉發功能,如下所示。
def forward(self, x):
x = self.pool(F.relu(self.conv1(x)))
x = F.relu(self.conv2(x))
x = x.view(-1, 16 *44*44)
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = self.fc3(x)
return x
要通過更改全連接層的維度來修復,請更改網絡的聲明如下。
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(3, 6, 5)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(6, 16, 5)
self.fc1 = nn.Linear(22*22*16, 120)
self.fc2 = nn.Linear(120, 84)
self.fc3 = nn.Linear(84, 10)
添加回答
舉報