[딥러닝] Pytorch 모델 구조 및 파라미터 확인

파이토치(Pytorch)를 이용한 모델 학습 시

사용되는 모델의 구조, 파라미터(parameters)의 이름 및 개수를 확인하고 싶을 때

 

아래의 파이썬 코드를 통해 확인 가능하다.

 

 

방법 1

for parameter in model.parameters():
    print(parameter)

 

 

방법 2

from torchsummary import summary

summary(model, (1,))    # inputs: model, input_size=(1,)
summary(model, (3,224,224))    # inputs: model, input_size=(3,224,224)

 

• 간단한 polynomial의 경우 입력의 차원은 (1,)
• 224x224 RGB 이미지의 경우 입력의 차원은 (3, 224, 224)