【发布时间】:2020-08-02 12:11:56
【问题描述】:
我正在使用 CNN 来识别 CiFar10 数据集中的图像,在添加 dropout 之前,cnn 的准确率达到了 58%,但在添加之后,它下降到了 52%。是不是网络没有过拟合?因为我怀疑是这样的。再添加两个 dropout 后,准确率上升到 55%,但我仍然对为什么它首先下降感到困惑。这是我的代码:
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(3, 12, 3, padding=1)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(12, 24, 3, padding=1)
self.conv3 = nn.Conv2d(24, 48, 3, padding=1)
self.conv4 = nn.Conv2d(48, 48, 3, padding=1)
self.dropout1 = nn.Dropout(p=0.2)
self.dropout2 = nn.Dropout(p=0.2)
self.dropout3 = nn.Dropout(p=0.3)
self.fc1 = nn.Linear(48 * 2 * 2, 120)
self.fc2 = nn.Linear(120, 84)
self.fc3 = nn.Linear(84, 10)
def forward(self, x):
x = self.pool(self.dropout1(F.relu(self.conv1(x))))
x = self.pool(self.dropout2(F.relu(self.conv2(x))))
x = self.pool(self.dropout3(F.relu(self.conv3(x))))
x = self.pool(F.relu(self.conv4(x)))
x = x.view(-1, 48 * 2 * 2)
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = self.fc3(x)
return x
【问题讨论】:
-
尝试绘制损失图以检查是否存在过拟合,因为 dropout 只有在过拟合时才有帮助。
标签: python pytorch conv-neural-network dropout