lightning/pytorch_lightning/plugins/precision
puhuk c58f84c176
Remove deprecated master_params attributes in PrecisionPlugin (#10372)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-11-08 02:42:03 +00:00
..
__init__.py Rename `TPUHalfPrecisionPlugin` to `TPUBf16PrecisionPlugin` (#10026) 2021-10-19 21:09:37 +00:00
apex_amp.py Remove `reinit_schedulers_properties` (#10271) 2021-11-01 11:05:29 +00:00
deepspeed_precision.py Clip before step (#10248) 2021-10-30 11:27:49 +01: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_precision.py Clip before step (#10248) 2021-10-30 11:27:49 +01:00
mixed.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
native_amp.py Clip before step (#10248) 2021-10-30 11:27:49 +01:00
precision_plugin.py Remove deprecated master_params attributes in PrecisionPlugin (#10372) 2021-11-08 02:42:03 +00: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