[PyTorch] torchinfo 사용방법 및 파라미터

파이토치의 모델 구조를 볼 수 있는 torchinfo라는 라이브러리가 있다.

아래와 비슷한 구조로 사용하며,

pip install torchinfo

from torchinfo import summary

model = LeNet()
summary(model,(1,1,3,3))

torchinfo 내에서 사용되는 파라미터는 다음과 같다.

summary(model, input_size, input_data, batch_dim, col_names, col_width, depth, device, dtypes, row_settings, verbose, **kwargs)