Carlos Mocholí
|
81d15c5986
|
Implement double optimizer closure for hook structure consistency (#10167)
|
2021-10-29 13:03:04 +00:00 |
|
Carlos Mocholí
|
48b6292cf0
|
Move optimizer step and clipping into the `PrecisionPlugin` (#10143)
|
2021-10-26 17:26:26 +02:00 |
|
Carlos Mocholí
|
b376799430
|
Minor fixes related to clipping (#10130)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
|
2021-10-25 16:40:22 +00:00 |
|
Adrian Wälchli
|
d41902883a
|
Update `optimizer_step` methods in accelerator and plugins (#10023)
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
|
2021-10-20 21:36:27 +01:00 |
|
Carlos Mocholí
|
ef5a12212a
|
Isolate optimizer step logic to the `PrecisionPlugin` (#10029)
|
2021-10-20 15:43:08 +00:00 |
|
Carlos Mocholí
|
53c62f63e8
|
Constrain IPU precision choices (#10030)
|
2021-10-20 00:52:01 +00:00 |
|
Jirka Borovec
|
6e124e7207
|
CI: precommit - docformatter (#8584)
* CI: precommit - docformatter
* fix deprecated
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-09-06 12:49:09 +00:00 |
|
Carlos Mocholí
|
a64cc37394
|
Replace `yapf` with `black` (#7783)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-07-26 13:37:35 +02:00 |
|
Carlos Mocholí
|
eb6d991218
|
Refactor plugins backward (#8328)
|
2021-07-08 16:02:09 +02:00 |
|
Sean Naren
|
96433d03ea
|
IPU Integration 5/5 (#7867)
* Initial changes
* Add broken example for now
* Fix reference
* Fix format
* Code runs
* Fixes
* Clear up files
* Add tests, helpers, fixes
* Small cleanups
* Refactors based on review
* Swap to special tests
* Add special tests
* Add source
* Cleanups
* Add logic to attach/detach model from devices
* Fixes for tests
* Fixes for tests
* Move earlier
* Cleanups
* Add check for nvcc
* Add tests, cleanups
* Fix errors
* fix
* Try condition
* Add missing annotation
* Clearer
* Clearer message
* Fix variable
* Cleanups
* Add comment
* CHANGELOG.md
* Add simple selection test
* Remove special=True to see what happens
* Fix test
* Update tests/accelerators/test_ipu.py
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
* Convert ipu_cores -> ipus
* Add typing, fail earlier
* simplify precision
* Add test, add helper
* fix accum
* Update pytorch_lightning/plugins/training_type/ipu.py
Co-authored-by: thomas chaton <thomas@grid.ai>
* Use stages
* Make sure warning message returned
* thorw error
* Add more tests, use fs
* add comment
* Clean
* Address feedback, add IPU tests
* Fixes
* Fix signature
* Add types
* Remove autoround
* Add docstring
* ipu_cores -> ipus
* Add test, remove unnecessary precision set
* Add optimizer test
* Add precision back with test
* Address code review
* Change to probs
* Move some of the asserts earlier
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: thomas chaton <thomas@grid.ai>
|
2021-06-11 15:07:04 +00:00 |