티스토리 뷰

기타

torchvision contribution #3

developer0hye 2022. 3. 28. 18:43

 

https://github.com/pytorch/vision/pull/5685

 

call _upcast to consider overflow for computing giou loss by developer0hye · Pull Request #5685 · pytorch/vision

giou loss is weak at overflow problem because it computes area of box. https://github.com/pytorch/vision/blob/main/torchvision/ops/boxes.py#L226-L242

github.com

 

torchvision에 기여! giou loss 계산과정중에 mixed precision으로 계산시 overflow가 날법한 부분이 있는데 이부분에서 overflow가 발생하지 않도록 upcasting하는 연산을 추가하였다.

'기타' 카테고리의 다른 글

[LeetCode] seanprashad list 부시기  (0) 2022.04.21
네이버랩스 뜨거운(불) 합격!  (6) 2022.04.02
[윈도우 cmd] rm -f  (0) 2022.03.14
20220309 C++ 공부해볼 것들  (0) 2022.03.09
20220206 블로그 누적 조회수 10,000 달성!  (0) 2022.02.06
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/01   »
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
글 보관함