분석시각화 대회 코드 공유 게시물은
내용 확인 후 좋아요(투표) 가능합니다.
포디블록 구조 추출 AI 경진대회
[public: 0.86392] timm, pytorch-lightning
timm, pytorch-lighting을 사용한 코드입니다.
원래 하나의 파이썬 파일로 작성한걸 노트북에 붙여넣어서 이런 모양이 되었습니다.
이 코드는 모델 파일의 크기가 100MB 정도의 작은 모델을 사용했습니다.
배치사이즈 32에서 VRAM 사용량은 8GB가 조금 안됩니다.
아마 모델을 큰 모델로 바꾸는것만으로도 점수가 더 올라가지 않을까 생각합니다.
코드 맨 밑에는 학습한 모델을 불러오는 방법을 적어두었으니 도움이 되길 바랍니다. 감사합니다.
모델 불러와서 따로 BCE로스걸어서 학습시켜서 Multi label한건가요?
네
파이토치 사용하니 학습속도가 너무 느리네요... 에폭당 보통 얼마나 걸리시나요?
코드에 사용된 maxvit_rmlp_tiny_rw_256모델로 RTX 3070에서 에포크당 5분, 총 50분 정도 걸렸습니다.
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io | 전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
감사합니다.🐰