7beed7cae6
Trainer cleanup ( #934 )
...
* Trainer cleanup
* update abstract
* remove ...
* remove __init__
* update mixin types
* update callbacks
* fix
* lower test acc
2020-02-27 16:21:14 -05:00
be244560b2
Callbacks [wip] ( #889 )
...
* Add callback system + associated test
* Add trainer and pl_module args to callback methods
* typing
* typo in docstring
* Switch to on_.*_start()
* fix on_test_start
* fix the mess after rebasing
2020-02-25 23:17:27 -05:00
edd4a87fb0
Refactor callbacks ( #776 )
...
* Refactor callbacks
* flake8
* Update docstrings
* Simplified callback, protected trainer
* .set_trainer() check
* update docs
* missed super().__ini__()
* Updated tests
* Use uppercase
* refine checkpoint callback tests
* Added test_begin() and test_end()
2020-02-16 00:03:05 -05:00
784a053793
Fix default ckpt path when logger exists ( #771 )
2020-02-01 13:36:50 -05:00
7deec2c14e
Move logger initialization ( #750 )
2020-01-26 09:42:57 -05:00
50881c0b31
Check early stopping metric in the beginning of the training ( #542 )
...
* Early stopping fix
* Update trainer.py
* Don't force validation sanity check
* fix tests
* update
* Added early_stopping check_metrics
* Updated docs
* Update docs
* Do not call early stopping when validation is disabled
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-01-23 11:12:51 -05:00
f72e354ee6
fixing TensorBoard ( #687 )
...
* flake8
* fix typo
* fix tensorboardlogger
drop test_tube dependence
* formatting
* fix tensorboard & tests
* upgrade Tensorboard
* test formatting separately
* try to fix JIT issue
* add tests for 1.4
2020-01-16 07:22:29 -05:00
88b750a018
default logger is now tensorboard ( #609 )
...
* refactor
* refactor
* refactor
* made tensorboard the default not test-tube
2020-01-14 14:40:41 -05:00
1d4b6be17b
rename trainer modules, drop `_mixin` ( #571 )
...
* rename trainer modules, drop _mixin
* fix imports
2019-12-04 11:39:14 -05:00
Jirka Borovec