반응형
pytorch
-
model.children() 이란?pytorch 2023. 11. 18. 01:59
요약 : forward에 구현되어 있는 모듈 아래에서는 6개 레이어로 구현되어 있으므로 6개가 나옴 model.children()은 PyTorch에서 제공하는 메서드로, 모델의 모든 하위 모듈을 반환합니다. 모델의 하위 모듈은 모델의 레이어, 컨볼루션 블록, 잔차 블록 등을 의미합니다. 예를 들어, 다음과 같은 모델이 있다고 가정합니다. import torch.nn as nn class MyModel(nn.Module): def __init__(self): super(MyModel, self).__init__() self.conv1 = nn.Conv2d(1, 32, 3) self.conv2 = nn.Conv2d(32, 64, 3) self.relu = nn.ReLU() self.maxpool = nn.M..