- 모델의 파라미터를 확인하기 위해 아래의 코드를 추가하셔서 사용하시면 됩니다.
- torchsummary는 keras의 summary 역할을 합니다.

```python
%pip install torchsummary ## model을 요약하기 위해 torchsummary 설치
from torchsummary import summary as summary_## 모델 정보를 확인하기 위해 torchsummary 함수 import

## 모델의 형태를 출력하기 위한 함수 
def summary_model(model,input_shape=(3,32,32)):
    model = model.cuda()
    summary_(model, input_shape) ## (model, (input shape))
```