PyTorch: Optimization & Training¶ https://github.com/pytorch/examples/tree/master/mnist In [1]: import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms import numpy as np In [2]: seed = 1 # 난수 고정 batch_size = 64 test_batch_size = 64 no_cuda = False In [3]: # gpu 사용? use_cuda = not no_cuda and torch.cuda.is_availa..

PyTorch Layer 이해하기¶ 예제 불러오기¶ In [6]: import torch from torchvision import datasets, transforms In [7]: import numpy as np import matplotlib.pyplot as plt %matplotlib inline In [8]: train_loader = torch.utils.data.DataLoader( datasets.MNIST('dataset', train=True, download=True, transform=transforms.Compose([ transforms.ToTensor() ])), batch_size=1) In [9]: image, label = next(iter(train_loader)) ..

https://github.com/pytorch/examples/tree/master/mnist PyTorch Data Preprocess¶ In [3]: import torch from torchvision import datasets, transforms Data Loader 부르기¶ 파이토치는 DataLoader를 불러 model에 넣음 In [4]: batch_size = 32 test_batch_size = 32 In [10]: train_loader = torch.utils.data.DataLoader( datasets.MNIST('dataset/',train = True,download = True, transform = transforms.Compose([ transforms.ToTenso..

PyTorch Basic¶ PyTorch 기초 사용법 In [1]: import numpy as np import torch In [2]: torch.__version__ Out[2]: '1.6.0' In [3]: nums = torch.arange(9) nums Out[3]: tensor([0, 1, 2, 3, 4, 5, 6, 7, 8]) In [4]: nums.shape Out[4]: torch.Size([9]) In [5]: type(nums) Out[5]: torch.Tensor In [6]: nums.numpy() Out[6]: array([0, 1, 2, 3, 4, 5, 6, 7, 8], dtype=int64) In [7]: nums.reshape(3,3) Out[7]: tensor([[0, ..