lightning/pytorch_lightning
William Falcon 0bd7780adc
Fixes CPU and hanging GPU crash (#2118)
* training batch clean up

* training batch clean up

* training batch clean up
2020-06-08 16:30:20 -04:00
..
callbacks Revert "Fixes EarlyStopping With Precision=16 (#1996)" (#2032) 2020-05-31 15:20:18 -04:00
core update hparams, allow OmegaConf (#2047) 2020-06-08 07:19:34 -04:00
loggers fix(wandb): use same logger on multiple training loops (#2055) 2020-06-02 18:46:02 -04:00
logging Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
metrics New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
overrides fix default arg (#1927) 2020-05-26 19:04:42 -04:00
profiler replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
pt_overrides Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
root_module Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
trainer Fixes CPU and hanging GPU crash (#2118) 2020-06-08 16:30:20 -04:00
utilities update hparams, allow OmegaConf (#2047) 2020-06-08 07:19:34 -04:00
__init__.py data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00