반응형
파이 토치 모델 summary에는 두가지 방법이 있습니다.
GRU 모델로 알아보겠습니다.
첫째는 단순히 model을 print하는 방법입니다.
print(model)
두번째는 torchsummary 모듈을 이용하는 것입니다.
from torchsummary import summary
summary(model, (1,1,26))
summary의 파라미터는 model과 인풋 사이즈입니다. 이때 배치와 시퀀스 길이는 1로 설정해야합니다. 이미지는 (채널 수, 이미지 사이즈)입니다.
반응형
'인공지능' 카테고리의 다른 글
| Pytorch CNN-LSTM 모델 설계 (0) | 2021.03.17 |
|---|---|
| pytorch class weight 주는법 (0) | 2021.03.16 |
| GRU VS LSTM (0) | 2021.03.15 |
| YOLO V5 사용하기 (0) | 2021.03.15 |
| YOLO v3 커스텀 데이터 셋 학습 (0) | 2021.01.05 |