Kaushik B
|
556879e5cf
|
Add support for devices flag to Trainer (#8440)
* Support devices flag to Trainer
|
2021-07-20 04:33:12 +00:00 |
|
Kaushik B
|
825c5dbe8c
|
Add support for (accelerator='cpu'|'gpu'|'tpu'|'ipu'|'auto') (#7808)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Ethan Harris <ewah1g13@soton.ac.uk>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: SeanNaren <sean@grid.ai>
|
2021-07-09 15:28:54 +00:00 |
|
Kaushik B
|
3f460b150a
|
Move parameter validation specific to TPU Training plugins (#7415)
* Move parameter validation specific to TPU Training plugins
* update docstring
|
2021-05-24 16:02:01 +00:00 |
|
Carlos Mocholí
|
8c0ea92af2
|
`TrainerState` refactor [5/5] (#7173)
* `TrainerState` refactor
* flake8
* Update finished check
* Test cleanup
* Fix tests
* Fixes
* Reorder
* flake8
* Update CHANGELOG
* Better docs
* Better docs
* Remove default
* Update tests
* Bad merge
|
2021-05-04 12:50:56 +02:00 |
|
Carlos Mocholí
|
40f80230fe
|
Remove `trainer.fit` return value [2/n] (#7237)
* `_fit_impl` refactor and types
* Fix return
* Remove return docstring
* Fixes
* Fixes
* Remove `trainer.fit` return value
* Update CHANGELOG
* flake8
* Undo results change
* Fix test
* Revert changes for a separate PR
* flake8
|
2021-04-28 19:11:32 +01:00 |
|
Kaushik B
|
f168a535ca
|
Add MpModelWrapper in TPU Spawn (#7045)
Co-authored-by: tchaton <thomas@grid.ai>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
|
2021-04-20 13:05:27 +00:00 |
|
Jirka Borovec
|
ac583781db
|
Refactor: Runif for TPU and Horovod 5/n (#6301)
* TPU
* horovod
* extra
* fix
* Apply suggestions from code review
Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>
* doc
Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>
|
2021-03-02 16:21:20 +00:00 |
|
Adrian Wälchli
|
fc9bb53e13
|
fix flake8 for new plugins (#5951)
* flake8
* fix cyclic import
* isort
|
2021-02-18 18:28:23 +00:00 |
|
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 |
|
Adrian Wälchli
|
a3d4e7c86a
|
move accelerator legacy tests (#5948)
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
|
2021-02-13 19:42:18 -05:00 |