Boostcamp AI Tech

[주말 실험 일지 - 토] Pstages - Semantic Segmentation

ju_young 2021. 10. 29. 12:56
728x90

이전에 24epoch을 돌린 모델을 불러와서 Dice Loss를 적용 -> FCN Head, OCR Head loss weight = 0.4 : 1.0으로 지정하고 class weight는 0.1, 1.0, 0.5, 1.5, 1.5, 1.5, 1.0, 1.25, 0.5, 2.0, 2.0으로 지정 -> 학습이 안됨 -> 정지 -> class weight를 지정해주었을 때 CE Loss가 잘 안줄어듬 -> class wieght를 삭제하고 CE + Dice Loss로 학습 -> miou가 53까지오르고 loss는 계속해서 내려감 -> 추가로 더 돌림

728x90