Pytorch_06_summary

이번 시간에는 Pytorch에 summary를 적용하는 방법에 대하여 알아보겠습니다.

summary

Pytorch에서 summary 기능을 사용하기 위해서는 keras와 다르게 summary를 불러와야합니다.

pip install torchsummary

프롬프트를 실행하고 가상환경을 활성화한 뒤 위의 코드를 통해 torchsummary를 설치해줍니다.

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

위와 같은 코드를 통해 model을 summary할 때에 아래와 같은 오류가 발생하는 것을 확인할 수 있습니다.

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

위와 같은 형식으로 수정해줍니다.

summary가 잘 실행되는 것을 확인할 수 있습니다.