lightning/pytorch_lightning/core
David Völgyes 651c25feb6
Fix for incorrect usage of detach(), cpu(), to() (#6216)
* Fix for incorrect detach/cpu calls (#6214)

* Fix incorrect use of detach(), to(), and cpu(), #6214

* Fix incorrect use of detach() and cpu(), #6214

* update pr

* add typing

* chlog

* more...

* revert on module

* update on comments

* revert changes on model

Co-authored-by: tchaton <thomas@grid.ai>
Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
2021-03-01 15:15:52 +00:00
..
__init__.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00
datamodule.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
decorators.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
grads.py formatting 5/n: Core (#5721) 2021-02-08 14:29:43 -05:00
hooks.py Add warnings to on_before/after_batch_transfer hooks (#6059) 2021-02-18 14:24:19 -05:00
lightning.py Prune deprecated hparams setter (#6207) 2021-02-27 13:24:50 +01:00
memory.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
optimizer.py Call `optimizer.zero_grad()` before backward inside closure in AutoOpt (#6147) 2021-03-01 14:36:46 +01:00
saving.py formatting 5/n: Core (#5721) 2021-02-08 14:29:43 -05:00
step_result.py Fix for incorrect usage of detach(), cpu(), to() (#6216) 2021-03-01 15:15:52 +00:00