lightning/pytorch_lightning/plugins/precision
Carlos Mocholí 44aed17aff
Remove duplicated native AMP + LBFGS check (#9748)
2021-09-29 13:14:03 +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 Remove duplicated native AMP + LBFGS check (#9748) 2021-09-29 13:14:03 +00:00
precision_plugin.py Remove unused `post_optimizer_step` (#9746) 2021-09-29 13:09:22 +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