Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/pytorch_lightning/utilities
History
Lezwon Castelino 69833dad5b
Added check to verify xla device is TPU (#3274) 
* tpu device check

* replaced with xmp spawn

* Revert "replaced with xmp spawn"

This reverts commit 6835380f

* replaced all instances of XLA_AVAILABLE

* moved inner_f to global scope

* made refactors

* added changelog

* added TPU_AVAILABLE variable

* fix codefactor issues

* removed form trainer and early stopping

* add TORCHXLA_AVAILABLE check

* added tests

* refactoring

* Update pytorch_lightning/utilities/xla_device_utils.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* updated function names

* fixed bug

* updated CHANGELOG.md

* added todo

* added type hints

* isort and black

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
 		2020-10-06 19:54:37 +02:00
..
__init__.py warn user when dropping unpicklable hparams (#2874) 2020-08-28 09:07:43 +02:00
apply_func.py implement fix and test (#3459) 2020-09-11 10:55:58 -04:00
argparse_utils.py ref: separate argparse (#3428) 2020-09-09 18:07:56 -04:00
cloud_io.py Use fsspec in load to resolve more paths/URLs from storage backends (#3692) 2020-10-03 13:29:03 -04:00
data.py Follow up of #2892 (#3202) 2020-08-27 15:28:29 -04:00
debugging.py Deprecate early_stop_callback Trainer argument (part 2) (#3845) 2020-10-04 17:36:47 -04:00
device_dtype_mixin.py Fix for PyTorch 1.7 CI (#3768) 2020-10-01 16:37:00 +02:00
device_parser.py ref: device parser (#3400) 2020-09-08 18:46:42 -04:00
distributed.py Fix log debug call (#3528) 2020-09-18 02:06:44 +05:30
exceptions.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
memory.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
model_utils.py ref: (results 1/n) enable tracking original metric when step and epoch are both true (#3685) 2020-09-27 22:08:31 -04:00
parsing.py Use `raise .. from ..` to explicitly chain exceptions (#3750) 2020-10-01 21:45:44 +02:00
seed.py Do not set PYTHONHASHSEED #2156 (#3745) 2020-09-30 08:38:24 -04:00
upgrade_checkpoint.py callback method for on_save_checkpoint (#2501) 2020-08-28 16:50:52 +02:00
warning_utils.py fix dp issues + update examples and test examples (#3618) 2020-09-23 00:19:46 -04:00
xla_device_utils.py Added check to verify xla device is TPU (#3274) 2020-10-06 19:54:37 +02:00