deeplearningzerotoall / PyTorch

Deep Learning Zero to All - Pytorch
1.03k stars 1.33k forks source link

lab-09_4_mnist_nn_deep.ipynb model 생성 부분 #46

Open JadeKim042386 opened 3 years ago

JadeKim042386 commented 3 years ago

linear를 1~5까지 만들어 놓고 이렇게 모델을 생성하는게 맞는건가요?

# model
model = torch.nn.Sequential(linear1, relu, linear2, relu, linear3).to(device)``