분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
[BASELINE] EfficientNet_b0을 활용한 분류 모델[public:0.67091]
안녕하세요😊 데이콘입니다.
이번 대회의 베이스라인 모델은 최정명님의 코드를 참고하여 작성되었습니다.
https://dacon.io/competitions/official/235870/codeshare/4171?page=3&dtype=recent
베이스라인은 참고용이며, 다양한 모델과 창의적인 방법으로 모델을 고도화 해주시기 바랍니다.
감사합니다.
실수로 삭제해 다시 올립니다.
Baseline 코랩으로 돌리려니
model=Network().to(deivce)부분에서
계속 램이 초과돼서 초기화되는데 해결 방법 있을까요?
img load 부분에서 이미지 사이즈를 512x512로 저장하기 때문에 램을 많이 잡아먹었던 것 같습니다.
사이즈 좀 줄여보니 baseline은 돌아가네요... ㅎㅎ;;
감사합니다.
import gc
gc.collect()
torch.cuda.empty_cache()
이용하시면 이미지 사이즈 변경 후에 GPU 캐시 비워주실수 있습니다.
저는 몰랐던 정보라 올려드려요!
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io | 전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
삭제된 댓글입니다