lightning/pytorch_lightning/plugins/precision
Sean Naren 1bab0a17a9
Fix torch bfloat import version (#9089)
2021-08-24 19:18:12 +00:00
..
__init__.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
apex_amp.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
deepspeed_precision.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
double.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
fully_sharded_native_amp.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
ipu_precision.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
mixed.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
native_amp.py Fix torch bfloat import version (#9089) 2021-08-24 19:18:12 +00:00
precision_plugin.py Add `pyupgrade` to `pre-commit` (#8557) 2021-07-26 14:38:12 +02:00
sharded_native_amp.py Add bfloat16 support to Lightning Trainer (#9049) 2021-08-24 09:47:21 +00:00
tpu_bfloat.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00