Feature/early stopping - Githubissues

aiorhiroki / farmer

farmer is an automated machine learning library.👨‍🌾

Apache License 2.0

12 stars 3 forks source link

Feature/early stopping #123

Closed sirn524 closed 4 years ago

sirn524 commented 4 years ago

Issueリンク

https://github.com/aiorhiroki/farmer/issues/117

実行したこと

keras.callbacksからEarlyStoppingの実装

できるようになること（ユーザ目線）

yamlにearly_stopping: True (またはyes)記載でEarlyStoppingの実行が可能。
yamlのpatience: (default:10)を操作することで、何Epochの間変更がない場合にStopするかを決めることが可能。
yamlのmonitor: (default:val_loss)を操作することで、どの指標を監視するかを変更可能。

動作確認

DL2上でのfarmer実行(patience:5の設定で、36Epoch目で学習終了）
Git Hub Actionsでの動作

参考情報

callbacksの最後にappendしないと実行されない可能性が高いため、trainer_task > _do_set_callbacks_taskの最終行に追加
callbacks追加に際しては、出来るだけ途中に記載いただきたい
Tensorflow のバージョン依存により、問題解消されている場合がある模様

sirn524 commented 4 years ago

Actionsを見ると、まだ問題解消していないように見えます。 Earlyすぎるstopping... 2epochで止まってしまうバグがどこかにありそうなので、関連する記事確認をして再度push試みます