lightning/pytorch_lightning/utilities
Justus Schock 742ddd81b3
make device property always return a device with index (#4851)
* make device property always return a device with index

* pep8

* Update test_dtype_device_mixin.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-11-27 00:49:51 +05:30
..
__init__.py Use high progress_bar_refresh_rate on Google Colab (#4654) 2020-11-24 02:13:33 +05:30
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 timeout for tpu check (#4340) 2020-11-01 01:04:25 +01:00