모델은 class로 구현한다.
class Net(nn.Module): def __init__(self) super(Net, self).__init__() 매개변수 정의.
재미있는 방식이다.
model = Net().to(DEVICE) 로 해당 디바이스에 할당하는 것까지가 모델만들기이다.
model = Net().to(DEVICE)