나의 첫번째 딥러닝 모델 만들기 - 모델 정의 및 인스턴스 생성

# 01. 라이브러리 불러오기 import torch import torch.nn as nn # 02. 모델 정의 및 인스턴스 생성 class SimpleModel(nn.Module): def __init__(self): super().__init__() self.linear = nn.Linear(1, 1) # 입력 차원 1, 출력 차원 1 def forward(self, x): return self.linear(x) # 모델 인스턴스 생성 model = SimpleModel() print(model, type(model))

* 성실한 태도, 작은 나의 땀과 노력이 모여 미래를 만듭니다. 정직, 성실 소중한 가치