DeepSpeedPrecisionPlugin¶
-
class
pytorch_lightning.plugins.precision.
DeepSpeedPrecisionPlugin
(precision)[source]¶ Bases:
pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin
Precision plugin for DeepSpeed integration.
-
backward
(model, closure_loss, optimizer, opt_idx, should_accumulate, *args, **kwargs)[source]¶ performs the actual backpropagation
- Parameters
- Return type
-