먼저 아나콘다 프롬프트(Anaconda prompt)에서 다음 명령을 실행하여 torchsummary 라이브러리를 설치합니다.

> pip install torchsummary

설치가 완료되었다면 torchsummary 라이브러리를 사용하여 모델의 네트워크 구조를 다시 확인해 봅시다.

코드 6-12 torchsummary 라이브러리를 이용한 모델의 네트워크 구조 확인

from torchsummary import summary
summary(model, input_size=(3,224,224)) ------ ①

① summary를 통해 모델의 네트워크 관련 정보를 확인할 수 있으며, 여기에서 사용되는 파라미터는 다음과 같습니다.

ⓐ 첫 번째 파라미터: 모델의 네트워크

ⓑ 두 번째 파라미터: 입력 값으로 (채널(channel), 너비(width), 높이(height))가 됩니다. 여기서는 (3, 244, 244)를 입력으로 지정합니다(이미 앞에서 244×244 크기의 이미지를 사용하겠다고 지정했습니다).