Jirka Borovec and William Falcon
64de57b09e
update checkpoint docs ( #1016 )
...
* update checkpoint docs
* fix tests
* fix tests
* formatting
* typing
* filename
* fix tests
* fixing tests
* fixing tests
* fixing tests
* unique name
* fixing
* fixing
* Update model_checkpoint.py
Co-authored-by: William Falcon <waf2107@columbia.edu >
2020-03-03 15:16:57 -05:00
William Falcon
6dae5698ef
fixes test issues on ddp ( #1017 )
...
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
* updated checkpoint docs
2020-03-02 21:50:38 -05:00
William Falcon
2a04be0386
No auto load weights ( #985 )
...
* remove autoload
* remove autoload
* added weights loading docs
* checkpoint loading saving docs
* checkpoint loading saving docs
* checkpoint loading saving docs
* docs (#1010 )
* remove autoload
* remove autoload
* added weights loading docs
* checkpoint loading saving docs
* checkpoint loading saving docs
* checkpoint loading saving docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
* docs
2020-03-02 17:12:22 -05:00
Jirka Borovec
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
Hadrien Mary
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
Vadim Bereznyuk
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
Vadim Bereznyuk
784a053793
Fix default ckpt path when logger exists ( #771 )
2020-02-01 13:36:50 -05:00
Vadim Bereznyuk
7deec2c14e
Move logger initialization ( #750 )
2020-01-26 09:42:57 -05:00
Vadim Bereznyuk and William Falcon
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
Jirka Borovec
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
William Falcon
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
Jirka Borovec
1d4b6be17b
rename trainer modules, drop _mixin ( #571 )
...
* rename trainer modules, drop _mixin
* fix imports
2019-12-04 11:39:14 -05:00