拡張機能として早期停止を実装する方法

Question

このthreadとして、関数（f：trainer - > bool）をTrainerのstop_trigerとして設定することで反復ループを止めることができます。 しかし、このように、私はstop_trigger =（（args.epoch、 '10'））を使用するLogReportのような他の拡張を使うことはできないと思います。

私の質問は、エクステンションとして早期停止を実装する方法と、エクステンションからトレーナーの反復を止める信号を送信する方法です。

ありがとうございました。

Answer 1

gist、 にサンプルコードを実装し、元のthreadの回答を更新しました。

stop_triggerは当初(args.epoch, '10')のようなタプル表記を使用していましたが、代わりに呼び出し可能オブジェクト（上記の例ではEarlyStoppingTrigger）を渡すように変更する必要があります。