부스트캠프 AI Tech 1기 [T1209 최보미]/P stage-1 (이미지분류)

Day5 학습정리

B1001101 2021. 4. 2. 23:57

1. 오늘의 목표

  • Stratified K-fold 적용하기

 

 

2. 목표를 달성하기 위해 한 행동

stackoverflow.com/questions/60883696/k-fold-cross-validation-using-dataloaders-in-pytorch에 나와있는 코드를 가져와서 k-fold를 적용해봤다. 내 코드에 적용하는 과정에서 몇 번의 에러가 있었지만 해결하고 정상적으로 작동시키는데 성공했다. 하지만 결과는 그렇게 좋지는 않았다.

 

3. 회고

오늘은 왠지 의욕이 떨어져서 대회에 열심히 참가하지 않고 k-fold밖에 시도를 안 해봤다. k-fold가 효과가 없었던 이유는 여러번 학습을 돌리다 보니 overfitting이 발생해서 그런 것 아닐까 생각한다. 주말동안 다른 여러가지 기법들도 적용해보고 Tensorboard도 사용해봐야겠다.