파이토치에서 model 정보(summary) 확인

파이토치에서 만들어진 모델 정보 확인하기

• keras에서는 model.summary()의 내장 함수를 이용하면 간단하게 모델 정보를 확인 가능합니다.
• 파이토치에서도 여러 가지 방법을 통해 정보를 확인할 수 있습니다.

1. print를 하면 summary가 출력된다

print(model)

2. torchinfo 통해 summary를 출력한다

!pip install torchinfo
from torchinfo import summary

summary(model)

3. torchsummary를 통해 summary출력 (input_size를 알아야만 출력 가능)

!pip install torchsummary
from torchsummary import summary

summary(model, input_size = (1,28,28), batch_size= 6)