월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

알고리즘 | CV | 분류 | 이상탐지 | Macro f1 score

  • moneyIcon Prize : 100만원 + ɑ
  • 1,392명 마감

 

[BASELINE] EfficientNet_b0을 활용한 분류 모델[public:0.67091]

2022.04.01 08:54 6,977 Views language

안녕하세요😊 데이콘입니다.

이번 대회의 베이스라인 모델은 최정명님의 코드를 참고하여 작성되었습니다.
https://dacon.io/competitions/official/235870/codeshare/4171?page=3&dtype=recent

베이스라인은 참고용이며, 다양한 모델과 창의적인 방법으로 모델을 고도화 해주시기 바랍니다.

감사합니다.

Code
로그인이 필요합니다
0 / 1000
민균
2022.04.02 17:42

삭제된 댓글입니다

민균
2022.04.02 17:50

실수로 삭제해 다시 올립니다. 
Baseline 코랩으로 돌리려니 
model=Network().to(deivce)부분에서 
계속 램이 초과돼서 초기화되는데 해결 방법 있을까요?

민균
2022.04.04 16:50

img load 부분에서 이미지 사이즈를 512x512로 저장하기 때문에 램을 많이 잡아먹었던 것 같습니다.
사이즈 좀 줄여보니 baseline은 돌아가네요... ㅎㅎ;; 

건국대김태형
2022.04.07 17:20

감사합니다.
import gc
gc.collect()
torch.cuda.empty_cache()
이용하시면 이미지 사이즈 변경 후에 GPU 캐시 비워주실수 있습니다.
저는 몰랐던 정보라 올려드려요!

이전 글
hydra + pytorch_lightning 기반 코드 공유
Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회
Likes 17
Views 2,858
Comments 1
3년 전
현재 글
[BASELINE] EfficientNet_b0을 활용한 분류 모델[public:0.67091]
Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회
Likes 24
Views 6,977
Comments 3
3년 전
다음 글
다음 글이 존재하지 않습니다.