我的电脑可以运行和学习,但精确度没有提高。列车损耗减小,但在列车集上测试时,具有随机精度。我在Keras中实现了相同的体系结构,准确率达到90%以上。我只是想知道我的结构是否有明显的问题。在这里:
def __init__(self):
super().__init__()
self.conv1 = nn.Conv2d(1, 16, 3)
self.conv2 = nn.Conv2d(16, 32, 3)
self.conv3 = nn.Conv2d(32, 64, 3)
self.fc1 = nn.Linear(5*5*64, 1024)
self.fc2 = nn.Linear(1024, 2048)
self.fc3 = nn.Linear(2048, 1)
def forward(self, x):
x = F.max_pool2d(F.relu(self.conv1(x)), (2, 2))
x = F.max_pool2d(F.relu(self.conv2(x)), (2, 2))
x = F.max_pool2d(F.relu(self.conv3(x)), (2, 2))
x = x.view(x.size(0), -1)
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = F.dropout(x, 0.5)
x = F.sigmoid(self.fc3(x))
return x
损失函数=nn.BCELoss()
目前没有回答
相关问题 更多 >
编程相关推荐