DACON Monthly Computer Vision Anomaly Detection Algorithm Competition

Algorithm | CV | Classification | Anomaly detection | Macro f1 score

 

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

2022.04.01 08:54 7,696 Views language

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

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

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

감사합니다.

Code
Login Required
0 / 1000
민균
2022.04.02 17:42

Deleted Comment

민균
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 캐시 비워주실수 있습니다.
저는 몰랐던 정보라 올려드려요!

Previous
hydra + pytorch_lightning 기반 코드 공유
Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회
Likes 17
Views 3,414
Comments 1
3yr ago
Current
[BASELINE] EfficientNet_b0을 활용한 분류 모델[public:0.67091]
Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회
Likes 24
Views 7,696
Comments 3
3yr ago
Next
No Next Post