PyTorch : Convolutional Neural Network

import libraries import torch import torch.nn as nn import torchvision.datasets as dset import torchvision.transforms as transforms from torch.autograd import Variable from torch.utils.data import DataLoader import matplotlib.pyplot as plt download data mnist_train = dset.MNIST("./", train=True, transform=transforms.ToTensor(), target_transform=None, download=True) 항목(item) print(mnist_train) # dataset.__getitem__(idx) image,label = mnist_train.

__getitem__(0) print(image.size(),label) # dataset[...