b35c472bb1
early stopping check_val_every_n_epoch fix ( #743 )
2020-01-24 18:18:51 -05:00
946aef6216
Added optimizer_idx to backward call ( #733 )
2020-01-24 18:03:07 -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
432a0bcd06
Issue #657 - Call on_train_end after early stopping ( #723 )
2020-01-21 16:48:30 -05:00
a2b20b46bc
remove unnecesarry gradient freeze/unfreeze for single optimizer setup ( #719 )
2020-01-21 08:09:27 -05:00
ea59a99426
update org paths & convert logos ( #685 )
...
* fix typos
* update org paths
* update links from READMe to docs
* add svg logo
* add svg logo-text
* update logos
* testing temp paths
* prune links from readme
* optimize imports
* update logo
* update paths in README
* missing imports
2020-01-20 14:50:31 -05:00
dac59bb8d3
replace obj.copy() with copy.copy(obj) ( #701 )
2020-01-17 08:10:05 -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
756c70a4a0
Clearer disable validation logic ( #650 )
...
* Clearer disable validation logic
* fix for fast_dev_run
* flake8 fix
* Test check fix
* update error message
2020-01-13 22:31:15 -05:00
0ae3dd9ed4
Fix GAN training. ( #603 )
...
* fix dangling gradients
make sure only the gradients of the current optimizer's paramaters are calculated in the training step.
* add note about multiple optimizer gradient update
* Update training_loop.py
2020-01-13 22:12:04 -05:00
12edc3099c
Fix the number of training batches used in the training loop ( #653 )
...
* Fix the number of processed training batches
* Fix tests
* fix tests
* fix tests
* One more attempt
* Fix another test
2020-01-05 14:37:09 -05:00
e2ee4ddbdb
Fix early stopping off by 2 (min_epochs) ( #617 )
...
* fix early stopping off by 2
* add min_epochs example in docs
2019-12-09 10:32:49 -08:00
b492e2b89e
Change nb to num in ABCs, comments, and tqdm logging ( #613 )
...
* Change nb to num in ABCs, comments, and tqdm logging
* Fix warnings text
* Make warnings one line
* Change num to number in comments
2019-12-09 04:40:26 -08:00
2baa80d626
Make sure train doesn't crash when called at max_epoch ( #608 )
2019-12-07 23:22:03 -05:00
2f01c03b38
Additional hooks ( #598 )
...
* Renamed `on_sanity_check_start` to `on_train_start` and added `on_train_end` to `ModelHooks`
* changed tests to use `on_train_start` instead of `on_sanity_check_start`
2019-12-07 08:52:06 -05:00
1051c189e1
Simplify variables: step, epoch, max_epochs, min_epochs ( #589 )
2019-12-07 08:50:21 -05:00
1d4b6be17b
rename trainer modules, drop `_mixin` ( #571 )
...
* rename trainer modules, drop _mixin
* fix imports
2019-12-04 11:39:14 -05:00
Vadim Bereznyuk