IPUPrecisionPlugin¶
- class lightning.pytorch.plugins.precision.IPUPrecisionPlugin(precision)[source]¶
Bases:
lightning.pytorch.plugins.precision.precision_plugin.PrecisionPlugin
Precision plugin for IPU integration.
Warning
This is an experimental feature.
- Raises
ValueError – If the precision is neither 16-mixed nor 32-true.
- backward(tensor, model, *args, **kwargs)[source]¶
Performs the actual backpropagation.
- Parameters
model¶ (
LightningModule
) – the model to be optimizedoptimizer¶ – current optimizer being used.
None
if using manual optimization*args¶ – Positional arguments intended for the actual function that performs the backward, like
backward()
.**kwargs¶ – Keyword arguments for the same purpose as
*args
.
- Return type