Early Stopping 구현 - Githubissues

boostcampaitech3 / level1-image-classification-level1-nlp-10

0 stars 4 forks source link

Early Stopping 구현 #8

Closed nonegom closed 2 years ago

nonegom commented 2 years ago

https://quokkas.tistory.com/37 다음 링크를 참조해서 EarlyStopping코드를 구현할 수 있었습니다.

pytorchtolls.py를 생성
train.py에서 EarlyStopping을 정의하고, early_stopping구현

우리 팀의 baseline코드의 경우 별도로 pth를 저장하기 때문에 save_checkpoint를 주석처리함.

git feature/changuk branch에서 확인 가능