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 |
|
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í
|
e8beceb631
|
Add `TPUPrecisionPlugin` (#10020)
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2021-10-19 17:48:57 +00:00 |