transformers/docs/source/main_classes
Colin Brochtrup 8ffc01a76a
Add early stopping callback to pytorch trainer (#8581)
* Add early stopping patience and minimum threshold metric must improve to prevent early stopping to pytorch trainer

* Add early stopping test

* Set patience counter to 0 if best metric not defined yet

* Make early stopping a callback. Add callback event for updating the best metric for early stopping callback to trigger on.

* Run make style

* make funciton name sensible

* Improve new argument docstring wording and hope that flakey CI test passes.

* Use on_evaluation callback instead of custom. Remove some debug printing

* Move early stopping arguments and state into early stopping callback

* Run make style

* Remove old code

* Fix docs formatting. make style went rogue on me.

* Remove copied attributes and fix variable

* Add assertions on training arguments instead of mutating them. Move comment out of public docs.

* Make separate test for early stopping callback. Add test of invalid arguments.

* Run make style... I remembered before CI this time!

* appease flake8

* Add EarlyStoppingCallback to callback docs

* Make docstring EarlyStoppingCallabck match other callbacks.

* Fix typo in docs
2020-11-23 17:25:35 -05:00
..
callback.rst Add early stopping callback to pytorch trainer (#8581) 2020-11-23 17:25:35 -05:00
configuration.rst Models doc (#7345) 2020-09-23 13:20:45 -04:00
logging.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
model.rst Refactoring the generate() function (#6949) 2020-11-03 16:04:22 +01:00
optimizer_schedules.rst Models doc (#7345) 2020-09-23 13:20:45 -04:00
output.rst [All Seq2Seq model + CLM models that can be used with EncoderDecoder] Add cross-attention weights to outputs (#8071) 2020-11-06 19:34:48 +01:00
pipelines.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
processors.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
tokenizer.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
trainer.rst Fix doc bug (#8500) 2020-11-12 11:47:23 -05:00