Commit Graph

7 Commits

Author SHA1 Message Date
Jirka Borovec 059eaecbb4
set xxx_AVAILABLE as protected (#5082)
* sett xxx_AVAILABLE as protected

* docs
2020-12-14 20:19:05 +05:30
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
Jirka Borovec 11e73ceaa6
fix import and typo in AMP (#4871)
* fix import and typo

* docs

* apex

* fix

* typo
2020-11-26 23:45:52 +01:00
Neven Miculinic 01402e3594
created minor doc fixes [ci skip] (#3958)
* created minor fixes

* adjusted the underline

* Update docs/source/amp.rst

* suggestion from code review

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2020-10-13 15:52:32 +02:00
William Falcon e68b949772
docs (#4003) 2020-10-08 15:54:52 -04:00
edenlightning 6dfa748ce3
Add video tutorials to docs (#3977)
* videos in trainer api

* videos in docs

* videos in docs

* videos in trainer api

* videos in docs

* videos in docs

* videos in docs

* videos in docs

* Update new-project.rst

* docs

* Update new-project.rst
2020-10-08 05:49:56 -04:00
Jeff Yang 951048a81e
docs: use ref for anchor links, fix a few typo (#3486) 2020-09-13 21:04:21 -04:00