lightning/pytorch_lightning/trainer
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
..
connectors Fix for incorrect usage of detach(), cpu(), to() (#6216) 2021-03-01 15:15:52 +00:00
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py simplify skip-if tests >> 0/n (#5920) 2021-03-01 12:17:09 +00:00
configuration_validator.py Add PredictLoop (#5752) 2021-02-16 17:11:56 -05:00
data_loading.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
deprecated_api.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
evaluation_loop.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
ignored_warnings.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
logging.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
model_hooks.py Replace .get_model() with explicit .lightning_module (#6035) 2021-02-18 15:59:54 +01:00
optimizers.py Prune deprecated metrics for 1.3 (#6161) 2021-02-24 11:09:01 +00:00
predict_loop.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
properties.py rename accelerator_backend -> accelerator (#6034) 2021-02-18 15:54:12 +00:00
states.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
supporters.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
trainer.py Fix for incorrect usage of detach(), cpu(), to() (#6216) 2021-03-01 15:15:52 +00:00
training_loop.py Fix for incorrect usage of detach(), cpu(), to() (#6216) 2021-03-01 15:15:52 +00:00
training_tricks.py Replace .get_model() with explicit .lightning_module (#6035) 2021-02-18 15:59:54 +01:00