lightning/tests/plugins/precision
Sean Naren bcbd9c359e
ShardedGradScaler should only be set for FP16 (#12915)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-04-28 17:44:31 +02:00
..
hpu Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
__init__.py
test_deepspeed_precision.py Add `bfloat16` support to `DeepSpeedStrategy` (#12508) 2022-04-07 22:18:51 +00:00
test_sharded_precision.py ShardedGradScaler should only be set for FP16 (#12915) 2022-04-28 17:44:31 +02:00
test_tpu_bf16_plugin.py