lightning/pytorch_lightning/utilities
chaton f2f4a49271 [bug-fix] Call transfer_batch_to_device in DDPlugin (#5195)
* hacking out

* update

* remove useless on_before_forward

* update

* remove overriden

* iremove os

* use on_before_forward

* resolve flake8

* add test

* update

* add single_process_per_device

* resolve flake8

* update

* resolve

* update

* update

* update

* add comment

* resolve bug with sharded

* update

* remove property

* update

* resolve test

* resolve bug

* update on comments

* update doc

* Update pytorch_lightning/core/hooks.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* update on comments

* Update pytorch_lightning/plugins/ddp_plugin.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* Update pytorch_lightning/plugins/ddp_plugin.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* resolve pep8

* add device_ids to pipe

* update on comments

* update

* resolve

* update

* update

* update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-109.ec2.internal>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>

(cherry picked from commit d510707bc9)
2021-01-26 14:28:45 +01:00
..
__init__.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
apply_func.py [bug-fix] Call transfer_batch_to_device in DDPlugin (#5195) 2021-01-26 14:28:45 +01:00
argparse.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
argparse_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
cloud_io.py Update cloud_io.py (#4936) 2020-12-04 11:00:06 +01:00
data.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
debugging.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
device_dtype_mixin.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
device_parser.py Raise TypeError instead of using asserting with condition of types. (#5536) 2021-01-19 07:39:43 -05:00
distributed.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
enums.py unify LightningEnum (#5389) 2021-01-12 08:56:07 -05:00
exceptions.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
imports.py clean avail. imports & enums (#5256) 2020-12-29 19:02:18 +01:00
memory.py [Fix] Move log value to cpu. (#4592) 2020-11-10 21:13:41 +00:00
model_helpers.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
model_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
parsing.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
seed.py Do not set PYTHONHASHSEED #2156 (#3745) 2020-09-30 08:38:24 -04:00
upgrade_checkpoint.py notices (#4118) 2020-10-13 07:18:07 -04:00
warning_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
warnings.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
xla_device.py mark todo exceptions (#5320) 2021-01-04 09:07:56 +01:00
xla_device_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00