분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
[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 캐시 비워주실수 있습니다.
저는 몰랐던 정보라 올려드려요!
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
Deleted Comment