티스토리 뷰

기타

20230423 DETR 맛보기중

developer0hye 2023. 4. 23. 22:49

https://github.com/facebookresearch/detr

 

GitHub - facebookresearch/detr: End-to-End Object Detection with Transformers

End-to-End Object Detection with Transformers. Contribute to facebookresearch/detr development by creating an account on GitHub.

github.com

 

 

8 V100 GPUs 로 300에폭 세팅으로 6일이 소요된다니... 살벌하다... 

 

프로젝트 틀만 파악하고 12~36 epoch 이내로 높은 수치의 AP를 달성해내는 Modern DETR 모델로 넘어가는 것도 생각해봐야겠다.

 

집 cpu가 i5이고, gpu가 3070인지라 집에서 간단히 돌려보기엔 무리일거같다...

 

돌려본다면  ResNet18로 백본을 대체하고, torch amp를 사용해서 조금이라도 시간을 단축시켜야 할 거 같다...

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

20230503 최고 월간 방문수 경신  (0) 2023.05.03
DINO SwinL 성능 해석시 주의점  (0) 2023.04.24
20230422 이번 연도 뭐할까  (0) 2023.04.22
20230415 대전 방문  (0) 2023.04.15
20230415 주간 방문수 1,103 을 기록하다!  (0) 2023.04.15
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
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
글 보관함