Ayberk Aydın
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
Vadim Bereznyuk
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
Adrian Wälchli
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
Elliot Waite
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
Nic Eggert
2baa80d626
Make sure train doesn't crash when called at max_epoch ( #608 )
2019-12-07 23:22:03 -05:00
schwobr
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
Elliot Waite
1051c189e1
Simplify variables: step, epoch, max_epochs, min_epochs ( #589 )
2019-12-07 08:50:21 -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
Ayberk Aydın