lightning/pytorch_lightning/core
Luis Perez 009e05d14f
[bugfix] Minor improvements to `apply_to_collection` and type signature of `log_dict` (#7851)
* minor fixeS

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-06-07 09:31:36 +01:00
..
__init__.py Deprecate `@auto_move_data` in favor of `trainer.predict` (#6993) 2021-04-19 14:53:21 +01:00
datamodule.py added stage param to LightningDataModule.setup example (#7483) 2021-05-11 23:43:22 +05:30
decorators.py Move parameter validation specific to TPU Training plugins (#7415) 2021-05-24 16:02:01 +00:00
grads.py Move grad_norm to a dedicated utilities file (#7292) 2021-04-30 09:19:22 -07:00
hooks.py move amp checkpoint state management to precision plugin (#7831) 2021-06-07 07:45:01 +00:00
lightning.py [bugfix] Minor improvements to `apply_to_collection` and type signature of `log_dict` (#7851) 2021-06-07 09:31:36 +01:00
memory.py Add `dataloader_idx` to batch transfer hooks (#6241) 2021-05-13 23:03:55 +05:30
optimizer.py refactor optimizer loop logic for manual and automatic optimization (#7526) 2021-05-17 14:42:01 +02:00
saving.py Fix csv extension check (#6436) 2021-04-08 01:16:31 +00:00