'Artificial Intelligence/Framework' 카테고리의 글 목록

[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.변수명 = ~ 을 사용할 경우 에러 발..

Artificial Intelligence/Framework 2022.09.16

일	월	화	수	목	금	토
			1	2	3	4
5	6	7	8	9	10	11
12	13	14	15	16	17	18
19	20	21	22	23	24	25
26	27	28	29	30	31

nsbg 🌞

Artificial Intelligence/Framework 1

티스토리툴바