Ayberk Aydın
a2b20b46bc
remove unnecesarry gradient freeze/unfreeze for single optimizer setup ( #719 )
2020-01-21 08:09:27 -05:00
Jirka Borovec
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
Z ZH
dac59bb8d3
replace obj.copy() with copy.copy(obj) ( #701 )
2020-01-17 08:10:05 -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
Vadim Bereznyuk
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
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