Data/Data Science
2018. 10. 4.
[Pytorch] MNIST DNN 코드 작성 & 공부
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 from __future__ import print_function import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from torch.autograd import Variable batch_size = 64 train_dataset = datasets.MNIST(root='./mnist_data/', train=True, transform=..
![[Pytorch] MNIST DNN 코드 작성 & 공부](http://i1.daumcdn.net/thumb/C480x320/?fname=https://blog.kakaocdn.net/dn/2hGit/btqSGkmzxPI/OKa7oRLdQpHn9jJTlXzqPk/img.png)