PyTorchでEarlyStoppingを実装する - Qiita

0.はじめに 今までKerasを使っていた人がいざpytorchを使うってなった際に、Kerasでは当たり前にあった機能がpytorchでは無い！というようなことに困惑することがあります。 KerasではcallbackとしてEarlyStoppingの機能が備わっていますが、Pytorchではデフォルトでこの機能は存在せず、自分で実装する必要があります。 今回はそれを実装したので共有しておきます。 参考：KerasのEarlyStopping 1.EarlyStoppingって何なのか？ 本記事を見るような方はすでにご存じかもしれませんが、一応説明します。 ・そもそも何Epoch学習を回せばいいのかなんて初見ではわからない ・Epoch数を重ねるたびにlossは下がるが、訓練データに過学習する可能性がある ・せっかくlossが下がっても、そのまま放置しすぎるとlossが上がっていってしま