PyTorch : Linear_mnist

import libraries import numpy as np import torch import torch.nn as nn import torch.optim as optim import torch.nn.init as init import torchvision.datasets as dset import torchvision.transforms as transforms from torch.utils.data import DataLoader from torch.autograd import Variable set hyperparameters batch_size = 16 learning_rate = 0.0002 num_epoch = 10 데이터 생성 download data mnist_train = dset.MNIST("./", train=True, transform=transforms.ToTensor(), target_transform=None, download=True) mnist_t...