lightning/pytorch_lightning/plugins/precision
Dusan Drevicky 1b06edf2f2
Add the `on_before_optimizer_step` hook ()
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-07-09 13:30:52 +02:00
..
__init__.py
apex_amp.py Add the `on_before_optimizer_step` hook () 2021-07-09 13:30:52 +02:00
deepspeed_precision.py Add the `on_before_optimizer_step` hook () 2021-07-09 13:30:52 +02:00
double.py
fully_sharded_native_amp.py
ipu_precision.py
mixed.py
native_amp.py Add the `on_before_optimizer_step` hook () 2021-07-09 13:30:52 +02:00
precision_plugin.py Add the `on_before_optimizer_step` hook () 2021-07-09 13:30:52 +02:00
sharded_native_amp.py Typing for accelerators and plugins () 2021-04-15 16:48:16 +00:00
tpu_bfloat.py