Kaushik B
|
56bc55db71
|
Update strategy flag in docs (#10000)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
|
2021-10-20 21:02:53 +05:30 |
Kaushik B
|
15434a9c35
|
Update torch_xla wheels installation link (#9436)
|
2021-09-10 20:00:22 +02:00 |
Jirka Borovec
|
0a71fe2859
|
CI: black docs (#8566)
* black docs
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-07-28 18:08:31 +02:00 |
Kaushik B
|
b069493b15
|
Add troubleshooting section for tpus (#8277)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
|
2021-07-12 13:29:02 +00:00 |
Kaushik B
|
b1a7b7e9bf
|
Add `tpuvm` section in TPU docs (#7714)
|
2021-05-26 12:41:00 +00:00 |
Kaushik B
|
22a266d8b8
|
Update TPU docs for installation (#6794)
|
2021-04-04 00:19:43 +05:30 |
Lezwon Castelino
|
d2cd7cb0f9
|
Add option for weight tying on TPU's (#5441)
* added on_post_move_to_device
* added tests
* docs and refactors
* Update tests/backends/test_tpu_backend.py
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* Update docs/source/tpu.rst
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* Update docs/source/tpu.rst
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* Update docs/source/tpu.rst
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* Update pytorch_lightning/core/decorators.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* Update pytorch_lightning/core/hooks.py
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
* moved weight sharing module back to test
updated tpu available
* add count to warning
* fix doctest
* import trainer in doctest
* import trainer in doctest
* do not test code as no TPU device
* param count to layer count
* formatting
* update docs
* update import
* update
* resolve tests
* remove legacy accelerator
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: tchaton <thomas@grid.ai>
Co-authored-by: Your Name <you@example.com>
|
2021-02-18 00:03:26 +00:00 |
William Falcon
|
b2950296d5
|
fixed TPU docs (#5958)
|
2021-02-15 13:58:15 +00:00 |
Jirka Borovec
|
dee5553b2b
|
move to Pages dir (#4869)
* folders
* common / advanced / extensions
* paths
* flake8
* isort
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
|
2021-01-26 15:07:07 -05:00 |