파이토치 모델의 개요(summary)를 어떻게 확인할 수 있나요?

[9bow (박정환)]

공식 홈페이지와 StackOverflow 등에서 자주 보이는 질문과 답변을 번역하고 있습니다.

다음 링크에서 원문을 함께 찾아보실 수 있습니다.

• 원문: python - How do I print the model summary in PyTorch? - Stack Overflow

---

질문

• 케라스(Keras)의 model.summary()처럼 모델의 개요(구조)를 보여주는 메소드는 없나요?

  Model Summary:
  ____________________________________________________________________________________________________
  Layer (type)                     Output Shape          Param #     Connected to                     
  ====================================================================================================
  input_1 (InputLayer)             (None, 1, 15, 27)     0                                            
  ____________________________________________________________________________________________________
  convolution2d_1 (Convolution2D)  (None, 8, 15, 27)     872         input_1[0][0]                    
  ____________________________________________________________________________________________________
  maxpooling2d_1 (MaxPooling2D)    (None, 8, 7, 27)      0           convolution2d_1[0][0]            
  ____________________________________________________________________________________________________
  flatten_1 (Flatten)              (None, 1512)          0           maxpooling2d_1[0][0]             
  ____________________________________________________________________________________________________
  dense_1 (Dense)                  (None, 1)             1513        flatten_1[0][0]                  
  ====================================================================================================
  Total params: 2,385
  Trainable params: 2,385
  Non-trainable params: 0