본문 바로가기

Artificial Intelligence/Framework1

[PyTorch] class 기반 모델 구현 방법 class ModelName(nn.Module): def __init__(self): super().__init__() # 코드 작성 부분 def forward(self): # 코드 작성 부분 nn.Module - 모든 신경망 모듈의 기본이 되는 클래스 - 모델 구현 시 이 클래스를 상속받아 사용해야 함(should) - nn.Module 클래스는 다른 nn.Module 안에서 사용될 수 있음 - Linear, Dropout, LayerNorm 등 모델 생성에 쓰이는 여러 기능 포함 __init__ - 신경망 구성 요소를 정의하기 위해 사용 - ✅ super().__init__(): 부모 클래스(nn.Module)를 호출하는 역할로 이 부분을 작성하지 않고 self.변수명 = ~ 을 사용할 경우 에러 발.. 2022. 9. 16.

이전 1 다음

티스토리툴바