pytorch學習(三):mnist數據集測試和訓練（二）卷積神經網絡的構建與訓練

1.構建卷積神經網絡 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(1, 10, kernel_size=5) self.conv2 = nn.Conv2d(10, 20, kernel_size=5)

>>阅读原文<<