Data/Data Science
2018. 10. 8.
[Pytorch] MNIST CNN 코드 작성 & 공부
앞에 import 한 내용, dataloader은 같은 내용이므로 생략한다. 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 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) self.mp = nn.MaxPool2d(2) self.fc = nn.Linear(320, 10) def forward(self, x): in_size = x.size(0) x = F.relu(self.mp(self.conv1(x))) x = F.relu(self...
![[Pytorch] MNIST CNN 코드 작성 & 공부](http://i1.daumcdn.net/thumb/C480x320/?fname=https://t1.daumcdn.net/cfile/tistory/99AAE83F5BBAE4510D)