lightning/pytorch_lightning/trainer
Nilansh Rajput 25c7465591
Enable gradients at train start (#2200)
* Enable gradients at train start

* Update training_loop.py

Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-06-17 10:52:58 -04:00
..
__init__.py [WIP] Rename overfit_pct to overfit_batches (and fix) and val_percent_check and test_percent_check (and fix) (#2213) 2020-06-17 08:03:28 -04:00
auto_mix_precision.py remove deprecated API for v0.8 (#2073) 2020-06-12 14:37:52 -04:00
callback_config.py Fix some pyright member access errors in training module (#2121) 2020-06-12 17:23:18 +02:00
callback_hook.py add on fit_start on fit_end hooks (#2217) 2020-06-17 07:37:16 -04:00
data_loading.py [WIP] Rename overfit_pct to overfit_batches (and fix) and val_percent_check and test_percent_check (and fix) (#2213) 2020-06-17 08:03:28 -04:00
deprecated_api.py remove deprecated API for v0.8 (#2073) 2020-06-12 14:37:52 -04:00
distrib_data_parallel.py hydra ddp support + generalized ddp gpu flags (#2212) 2020-06-17 10:33:25 -04:00
distrib_parts.py Finish Ananthsub patch 1 (enable prepare_data from correct processes). clarify local vs global rank (#2166) 2020-06-13 12:00:14 -04:00
evaluation_loop.py [WIP] Rename overfit_pct to overfit_batches (and fix) and val_percent_check and test_percent_check (and fix) (#2213) 2020-06-17 08:03:28 -04:00
ignored_warnings.py Dim 0 warning (#256) 2019-09-26 13:20:54 -04:00
logging.py Finish Ananthsub patch 1 (enable prepare_data from correct processes). clarify local vs global rank (#2166) 2020-06-13 12:00:14 -04:00
lr_finder.py Fix some pyright member access errors in training module (#2121) 2020-06-12 17:23:18 +02:00
model_hooks.py Fix some pyright member access errors in training module (#2121) 2020-06-12 17:23:18 +02:00
optimizers.py Fix user warning produced by apex + scheduler combination (#1873) 2020-05-22 07:19:37 -04:00
seed.py Fix some pyright member access errors in training module (#2121) 2020-06-12 17:23:18 +02:00
supporters.py Fix some pyright member access errors in training module (#2121) 2020-06-12 17:23:18 +02:00
trainer.py [WIP] Rename overfit_pct to overfit_batches (and fix) and val_percent_check and test_percent_check (and fix) (#2213) 2020-06-17 08:03:28 -04:00
training_io.py test: save hparams to yaml (#2198) 2020-06-16 06:34:55 -04:00
training_loop.py Enable gradients at train start (#2200) 2020-06-17 10:52:58 -04:00
training_tricks.py Revert "Misleading exception raised during batch scaling (#1973)" (#2219) 2020-06-17 08:01:53 -04:00