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

..

__init__.py

Sphinx generated documentation (#521)

2019-11-28 12:48:55 -05:00

auto_mix_precision.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00

callback_config.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00

data_loading.py

Fix percent_checks (#649)

2020-01-05 14:36:06 -05:00

distrib_data_parallel.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00

distrib_parts.py

Docs and Tests for "gpus" Trainer Argument (#593)

2019-12-07 08:48:45 -05:00

evaluation_loop.py

Change nb to num in ABCs, comments, and tqdm logging (#613)

2019-12-09 04:40:26 -08:00

ignored_warnings.py

Dim 0 warning (#256)

2019-09-26 13:20:54 -04:00

logging.py

Change nb to num in ABCs, comments, and tqdm logging (#613)

2019-12-09 04:40:26 -08:00

model_hooks.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00

trainer_io.py

Abstract Mixin classes (#572)

2019-12-04 10:57:32 -05:00

trainer.py

Change nb to num in ABCs, comments, and tqdm logging (#613)

2019-12-09 04:40:26 -08:00

training_loop.py

Fix GAN training. (#603)

2020-01-13 22:12:04 -05:00

training_tricks.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00