lightning/docs/source
Lezwon Castelino 12cb9942a1
Tpu save (#4309)
* convert xla tensor to cpu before save

* move_to_cpu

* updated CHANGELOG.md

* added on_save to accelerators

* if accelerator is not None

* refactors

* change filename to run test

* run test_tpu_backend

* added xla_device_utils to tests

* added xla_device_utils to test

* removed tests

* Revert "added xla_device_utils to test"

This reverts commit 0c9316bb

* fixed pep

* increase timeout and print traceback

* lazy check tpu exists

* increased timeout
removed barrier for tpu during test
reduced epochs

* fixed torch_xla imports

* fix tests

* define xla utils

* fix test

* aval

* chlog

* docs

* aval

* Apply suggestions from code review

Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: chaton <thomas@grid.ai>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>
2020-12-02 13:05:11 +00:00
..
_images add svg logo [SKIP CI] (#3988) 2020-10-21 19:11:31 +02:00
_static Callback docs with autosummary (#3908) 2020-10-06 17:28:45 -04:00
_templates logger docs and api docs (#3950) 2020-11-13 20:35:54 +05:30
accelerators.rst Accelerator docs (#4583) 2020-11-08 17:24:41 -05:00
amp.rst Tpu save (#4309) 2020-12-02 13:05:11 +00:00
api_references.rst logger docs and api docs (#3950) 2020-11-13 20:35:54 +05:30
asr_nlp_tts.rst Update to NeMo Documentation (#4328) 2020-10-25 06:57:04 -04:00
bolts.rst Documentation Fixes [skip ci] (#3955) 2020-10-07 16:10:41 -04:00
callbacks.rst update documentation for callbacks (#4253) 2020-10-20 16:01:08 +01:00
child_modules.rst Drop all result docs. Make the separation between flow and logging clear (#3744) 2020-09-30 08:31:16 -04:00
cloud_training.rst docs (#4096) 2020-10-12 12:15:33 -04:00
community_examples.rst Add lightning-geometric (#4771) 2020-11-20 08:45:04 +01:00
conf.py Tpu save (#4309) 2020-12-02 13:05:11 +00:00
converting.rst Fixes broken LM links (#4117) 2020-10-13 06:42:33 -04:00
datamodules.rst clean up docs (#4095) 2020-10-12 11:56:16 -04:00
debugging.rst Model summary: add 1 decimal place (#4745) 2020-11-20 23:22:21 +01:00
early_stopping.rst docs (#4003) 2020-10-08 15:54:52 -04:00
fast_training.rst created minor doc fixes [ci skip] (#3958) 2020-10-13 15:52:32 +02:00
governance.rst add s-rog to core (#4017) 2020-10-09 13:41:57 +05:30
hyperparameters.rst Update old "module_arguments" and "hparams" references in docs (#4417) 2020-11-03 12:13:10 +01:00
index.rst [Example] Add Pytorch Geometric Example (#4568) 2020-11-18 20:03:55 +00:00
introduction_guide.rst [docs] distributed_backend -> accelerator (#4429) 2020-10-30 00:45:24 +06:30
lightning_module.rst optimizer clean up (#4658) 2020-12-01 00:09:46 +00:00
loggers.rst Adds shortcut for path to log (#4573) 2020-11-08 12:16:22 -05:00
logging.rst update logging docs and decorators (#4431) 2020-12-01 11:35:00 +05:30
lr_finder.rst Fixes broken LM links (#4117) 2020-10-13 06:42:33 -04:00
metrics.rst [Metrics] Unification of FBeta (#4656) 2020-11-23 09:44:35 +01:00
multi_gpu.rst Sharded DDP Docs (#4920) 2020-12-02 11:54:46 +00:00
multiple_loaders.rst Minor formatting & grammar fixes in docs [CI SKIP] (#3952) 2020-10-07 14:25:52 -04:00
new-project.rst Fix codes in 'Lightning in 2 steps' docs (#4894) 2020-11-30 14:29:49 +01:00
optimizers.rst optimizer clean up (#4658) 2020-12-01 00:09:46 +00:00
performance.rst Sharded DDP Docs (#4920) 2020-12-02 11:54:46 +00:00
plugins.rst Enable custom apex and amp plugins (#4355) 2020-10-25 17:11:07 -04:00
production_inference.rst docs: use ref for anchor links, fix a few typo (#3486) 2020-09-13 21:04:21 -04:00
profiler.rst Add labels to sphinx docs (#2964) 2020-08-13 18:56:51 -04:00
pytorch_ecoystem.rst [Example] Add Pytorch Geometric Example (#4568) 2020-11-18 20:03:55 +00:00
rapid_prototyping_templates.rst docs fixes (#4080) 2020-10-11 13:12:19 -04:00
sequences.rst [CI SKIP] Sequential data & TPU support docs fix (#3956) 2020-10-07 16:11:03 -04:00
single_gpu.rst docs: use ref for anchor links, fix a few typo (#3486) 2020-09-13 21:04:21 -04:00
slurm.rst [docs] distributed_backend -> accelerator (#4429) 2020-10-30 00:45:24 +06:30
style_guide.rst fixed typos in style guide (#4181) 2020-10-16 22:21:47 +02:00
test_set.rst Fixes broken LM links (#4117) 2020-10-13 06:42:33 -04:00
tpu.rst [DOC] Clarify `tpu_cores` training. (#4475) 2020-11-02 15:23:24 +00:00
trainer.rst Tpu save (#4309) 2020-12-02 13:05:11 +00:00
training_tricks.rst Revert "Remove limitation of batch scaler (#4006)" (#4040) 2020-10-09 21:03:23 -04:00
transfer_learning.rst Small documentation fix (#3589) 2020-09-22 09:45:01 -04:00
weights_loading.rst Update old "module_arguments" and "hparams" references in docs (#4417) 2020-11-03 12:13:10 +01:00