https://pytorch.org/tutorials/beginner/basics/quickstart_tutorial.html

https://colab.research.google.com/drive/1NAXY6ELyNxDHt13nw_cHnBePrsdQTuH-?usp=sharing

샘플 데이터셋 다운받기

torch.utils.data.Dataset
torch.utils.data.DataLoader

<aside> 📌

Wrapper class 래퍼 클래스

</aside>

<aside> 📌

Parrallel 병렬 처리

Creating Models

class NeuralNetwork(nn.Module): # nn.Module 상속
    def __init__(self): # 클래스 초기화
        super().__init__() # 부모 클래스 초기화
        self.flatten = nn.Flatten() # 평탄화
        self.linear_relu_stack = nn.Sequential( 
		        # nn.Linear(input data size, output data size)
		        # == y = ax + b 선형 변환 (직선만 생성)
            nn.Linear(784, 512), # input size == flattened data size
            nn.ReLU(), # activatation 함수: 비선형 변환 추가 (곡선 추가)
            nn.Linear(512, 512),
            nn.ReLU(),
            nn.Linear(512, 10) # 아웃풋 클래스(y 라벨)는 총 10개
        )
# 위 NeuralNetwork 클래스에 이어서...
    def forward(self, x): # 실제로 외부에서 호출하는 함수
        x = self.flatten(x)
        logits = self.linear_relu_stack(x)
        return logits

model = NeuralNetwork().to("cpu")