IPUPrecisionPlugin¶
- class pytorch_lightning.plugins.precision.IPUPrecisionPlugin(precision)[source]¶
Bases:
pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin
- backward(model, *args, **kwargs)[source]¶
Performs the actual backpropagation.
- Parameters
model¶ (
LightningModule
) – the model to be optimizedclosure_loss¶ – the loss value obtained from the closure
optimizer¶ – current optimizer being used.
None
if using manual optimization
- Return type