"PyTorch:모델"의 두 판 사이의 차이

Pywiki
둘러보기로 가기 검색하러 가기
(새 문서: == 작성 == 모델은 class로 구현한다.<syntaxhighlight lang="python"> class Net(nn.Module): def __init__(self) super(Net, self).__init__() 매개변수 정...)
 
(차이 없음)

2022년 2월 15일 (화) 11:21 기준 최신판

작성[편집 | 원본 편집]

모델은 class로 구현한다.

class Net(nn.Module):
    def __init__(self)
        super(Net, self).__init__()
        매개변수 정의.

재미있는 방식이다.

model = Net().to(DEVICE) 로 해당 디바이스에 할당하는 것까지가 모델만들기이다.