pytorch model summary

파이 토치 모델 summary에는 두가지 방법이 있습니다.

GRU 모델로 알아보겠습니다.

첫째는 단순히 model을 print하는 방법입니다.

print(model)

두번째는 torchsummary 모듈을 이용하는 것입니다.

from torchsummary import summary
summary(model, (1,1,26))

summary의 파라미터는 model과 인풋 사이즈입니다. 이때 배치와 시퀀스 길이는 1로 설정해야합니다. 이미지는 (채널 수, 이미지 사이즈)입니다.