분류 전체보기 (81) 썸네일형 리스트형 [PyTorch] torch.nn.ModuleList()란? Tensorflow만 주로 사용하다가 pytorch를 사용한 모델을 베이스로 개선하려니 모르는 것 투성이라서 하나씩 정리해보기로 한다! ModuleList — PyTorch 1.10.0 documentation Shortcuts pytorch.org [ModuleList] 모듈을 리스트 형태로 저장하는 것을 의미한다. 파이썬의 일반적인 리스트처럼 인덱스로 접근이 가능하다. 공식 홈페이지에 있는 사용 예제를 살펴보자. class MyModule(nn.Module): def __init__(self): super(MyModule, self).__init__() self.linears = nn.ModuleList([nn.Linear(10, 10) for i in range(10)]) def forward(se.. 이전 1 ··· 8 9 10 11 다음