lightning/pytorch_lightning/utilities
Jirka Borovec 442d57f1e9
simplify imports xla / TPU (#4872)
* xla

* tpu

* fix

* fix

* flake8
2020-11-27 00:37:48 +01:00
..
__init__.py simplify imports xla / TPU (#4872) 2020-11-27 00:37:48 +01:00
apply_func.py Fix torchtext data to gpu (#4785) 2020-11-24 00:27:14 -05:00
argparse_utils.py feature: Allow str arguments in Trainer.profiler (#3656) 2020-10-27 16:27:16 +05:30
cloud_io.py Load checkpoint from Bytes (#4314) 2020-10-23 11:29:13 -04:00
data.py Revert "Remove limitation of batch scaler (#4006)" (#4040) 2020-10-09 21:03:23 -04:00
debugging.py [test] Accumulated gradient optimization tests (#4477) 2020-11-02 23:44:11 +00:00
device_dtype_mixin.py make device property always return a device with index (#4851) 2020-11-27 00:49:51 +05:30
device_parser.py Set correct device ids in DDP [wip] (#4297) 2020-10-24 17:33:47 -04:00
distributed.py Horovod: fixed early stopping and added metrics aggregation (#3775) 2020-11-05 12:52:02 -05:00
exceptions.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
memory.py [Fix] Move log value to cpu. (#4592) 2020-11-10 21:13:41 +00:00
model_utils.py add test for model hooks (#4010) 2020-10-20 13:33:46 +01:00
parsing.py allow decorate model init with saving hparams (#4662) 2020-11-16 11:02:26 +01: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 notices (#4118) 2020-10-13 07:18:07 -04:00
xla_device_utils.py simplify imports xla / TPU (#4872) 2020-11-27 00:37:48 +01:00