lightning/pytorch_lightning/plugins/precision
four4fish 44cd412e91
Remove precision_plugin pre_dispatch() method (#10887)
* Remove precision_plugin pre_dispatch() method

* update changelog
2021-12-01 18:42:17 -08:00
..
__init__.py Remove the "_precision" suffix from some precision plugin files (#10052) 2021-11-19 17:37:39 +00:00
apex_amp.py 2/n Move Precision Plugin into strategy - move optimizer related logics (#10596) 2021-11-30 08:31:23 +00:00
deepspeed.py Improve typing for plugins (#10742) 2021-11-26 20:14:58 +00:00
double.py Introduce `PrecisionPlugin.forward_context()` (#9988) 2021-10-18 12:58:19 +00:00
fully_sharded_native_amp.py Pass the scaler as an input to `NativeMixedPrecisionPlugin` (#10055) 2021-10-28 14:13:53 +00:00
ipu.py Remove the "_precision" suffix from some precision plugin files (#10052) 2021-11-19 17:37:39 +00:00
mixed.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
native_amp.py Improve typing for plugins (#10742) 2021-11-26 20:14:58 +00:00
precision_plugin.py Remove precision_plugin pre_dispatch() method (#10887) 2021-12-01 18:42:17 -08:00
sharded_native_amp.py Pass the scaler as an input to `NativeMixedPrecisionPlugin` (#10055) 2021-10-28 14:13:53 +00:00
tpu.py Implement double optimizer closure for hook structure consistency (#10167) 2021-10-29 13:03:04 +00:00
tpu_bf16.py Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026) 2021-10-19 21:09:37 +00:00