강의

멘토링

커뮤니티

Inflearn コミュニティ Q&A

k3y3j39254 のプロフィール画像
k3y3j39254

投稿した質問数

[PyTorch] 実戦人工知能へとつながるディープラーニング - 基礎から論文実装まで

早期終了 (Early stopping)

early stopping 코드 문의

解決済みの質問

作成

·

316

0

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

    if val_loss < early_stopping_loss:
        torch.save(resnet.state_dict(), PATH)
        early_stopping_train_loss = train_loss
        early_stopping_val_loss = val_loss
        early_stopping_epoch = epoch

강사님 안녕하세요.

위 코드에 궁금한 점이 있어서 질문드립니다.

위 코드의 4번째 줄에서 아래와 같이 early_stopping_loss 변수를 업데이트 해줘야 하는게 아닌지 궁금합니다.

early_stopping_loss = val_loss

지금 코드 상으로는 early_stopping_loss가 업데이트 되는 부분이 없어보여서요. 지금 코드로는 모든 epoch에서 if 문에 들어가는것이 아닐지 질문드립니다!

 

감사합니다.

 

python머신러닝딥러닝인공신경망pytorch

回答 1

0

dlbro님의 프로필 이미지
dlbro
インストラクター

안녕하세요.

제보 감사합니다.

오기가 맞습니다 ㅠㅠ

early_stopping_val_loss를 early_stopping_loss으로 해야합니다.

파일은 업데이트 하였습니다.

 

연 말 잘 보내세요 :)

k3y3j39254 のプロフィール画像
k3y3j39254

投稿した質問数

質問する