69833dad5b
* 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> |
||
|---|---|---|
| .. | ||
| data/horovod | ||
| __init__.py | ||
| test_amp.py | ||
| test_cpu.py | ||
| test_gpu.py | ||
| test_grad_norm.py | ||
| test_hooks.py | ||
| test_horovod.py | ||
| test_hparams.py | ||
| test_onnx.py | ||
| test_restore.py | ||
| test_sync_batchnorm.py | ||
| test_torchscript.py | ||
| test_tpu.py | ||