lightning/pytorch_lightning/plugins/precision
Carlos Mocholí 6892d533ea
Run plugin closure before `on_before_optimizer_step` [1/2] (#9288)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-09-07 11:52:20 +00:00
..
__init__.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
apex_amp.py Run plugin closure before `on_before_optimizer_step` [1/2] (#9288) 2021-09-07 11:52:20 +00:00
deepspeed_precision.py Run plugin closure before `on_before_optimizer_step` [1/2] (#9288) 2021-09-07 11:52:20 +00:00
double.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
fully_sharded_native_amp.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
ipu_precision.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
mixed.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
native_amp.py Run plugin closure before `on_before_optimizer_step` [1/2] (#9288) 2021-09-07 11:52:20 +00:00
precision_plugin.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
sharded_native_amp.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
tpu_bfloat.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00