lightning/pytorch_lightning/plugins/precision
Sean Naren 39231aee1a
[Fix] Call clip gradients if clip val greater than 0 (#6330)
* Call clip gradients if clip val greater than 0

* format

* Format

* Move to top of file
2021-03-04 19:45:58 +00:00
..
__init__.py DeepSpeed Integration (#5954) 2021-02-17 15:23:42 -05:00
apex_amp.py Simplify test for AMP plugins (#6311) 2021-03-03 08:56:57 +01:00
deepspeed_precision.py simplify skip-if tests >> 0/n (#5920) 2021-03-01 12:17:09 +00:00
mixed.py Add mypy typing to precision plugins. (#6149) 2021-02-26 14:27:16 +01:00
native_amp.py Simplify test for AMP plugins (#6311) 2021-03-03 08:56:57 +01:00
precision_plugin.py Add mypy typing to precision plugins. (#6149) 2021-02-26 14:27:16 +01:00
sharded_native_amp.py [Fix] Call clip gradients if clip val greater than 0 (#6330) 2021-03-04 19:45:58 +00:00
tpu_bfloat.py Add mypy typing to precision plugins. (#6149) 2021-02-26 14:27:16 +01:00