[PyTorch] torch.exp 와 auto mixed precision
모델을 구성하는 레이어(혹은 연산) 중 torch.exp 를 포함하는 경우 auto mixed precision과 충돌이 발생할 수 있습니다. 이는 torch.exp의 output data type이 auto mixed precision과 호환되도록 자동으로 변환되지 않아 발생하는 문제입니다. (추후 업데이트에서 해결될 문제로 보이긴 합니다.) 이런 경우 torch.exp 의 output의 data type을 입력 텐서 x의 data type으로 재변환시키는 과정이 필요합니다. output = torch.exp(x).type(x.dtype)
Deep Learning/PyTorch
2021. 1. 19. 00:32
YOLOv3 구현을 위해 참고할 자료들
github.com/AlexeyAB/darknet/issues/232 Resizing : keeping aspect ratio, or not · Issue #232 · AlexeyAB/darknet Hi, In your implementation of Darknet, when resizing the image, the aspect ratio of the image is not kept in the function get_image_from_stream_resize: darknet/src/image.c Line 627 in 5a2efd5 image... github.com github.com/pjreddie/darknet/issues/1354 Cross Entropy for YOLOv3 · Issue #1..
Deep Learning
2021. 1. 16. 20:50
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 파이참
- 위상 정렬 알고리즘
- 백준
- 인공지능을 위한 선형대수
- 가장 긴 증가하는 부분 수열
- 순열
- 백준 1766
- 문제집
- 조합
- FairMOT
- 단축키
- 이분탐색
- MOT
- Lowest Common Ancestor
- PyCharm
- 백준 11053
- 백준 11437
- 자료구조
- ㅂ
- C++ Deploy
- cosine
- LCA
- 백트래킹
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | |||
5 | 6 | 7 | 8 | 9 | 10 | 11 |
12 | 13 | 14 | 15 | 16 | 17 | 18 |
19 | 20 | 21 | 22 | 23 | 24 | 25 |
26 | 27 | 28 | 29 | 30 | 31 |
글 보관함