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

fix metric name to work with default earlystopping (#628)

2019-12-14 20:23:44 -08:00

decorators.py

Refactor: name modules (#548)

2019-11-26 22:39:18 -05:00

grads.py

Refactor: name modules (#548)

2019-11-26 22:39:18 -05:00

hooks.py

Additional hooks (#598)

2019-12-07 08:52:06 -05:00

lightning.py

Fix GAN training. (#603)

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

memory.py

Allow for multiple example inputs when creating summary (#543)

2019-12-09 04:42:07 -08:00

model_saving.py

CI pass (#671)

2020-01-13 22:09:47 -05:00

root_module.py

CI pass (#671)

2020-01-13 22:09:47 -05:00

saving.py

Refactor: name modules (#548)

2019-11-26 22:39:18 -05:00