반응형
GRU는 각 셀에서 cell state를 전달하지 않아 LSTM보다 연산량이 적어 속도가 빠른 장점이 있습니다.
그렇다면 속도를 CPU와 GPU환경에서 비교해보겠습니다.
실행한 테스크는 포즈 디텍션이며 pytorch 프레임 워크를 이용했습니다. 러닝 타임의 평균값을 내기 위해 for문을 이용하여 2번 실행했습니다. 모델은 bidirectional LSTM과 GRU입니다.
CPU환경
LSTM - 0.2s
GRU - 0.06s
GPU 환경
LSTM - 0.005s
GRU - 0.006s
결론적으로 CPU환경에서 GRU를 사용하는 편이 LSTM 보다 3배정도 빠르며 GPU환경에선 다소 차이가 안나는 것으로 보입니다.
반응형
'인공지능' 카테고리의 다른 글
| pytorch class weight 주는법 (0) | 2021.03.16 |
|---|---|
| pytorch model summary (0) | 2021.03.16 |
| YOLO V5 사용하기 (0) | 2021.03.15 |
| YOLO v3 커스텀 데이터 셋 학습 (0) | 2021.01.05 |
| YOLO V3 원하는 객체만 바운더리 박스 칠하기 (0) | 2021.01.05 |