diff --git a/docs/source/common/optimizers.rst b/docs/source/common/optimizers.rst index ca10dc97f1459..12e9c6925e7fd 100644 --- a/docs/source/common/optimizers.rst +++ b/docs/source/common/optimizers.rst @@ -49,7 +49,7 @@ Here is a minimal example of manual optimization. # Important: This property activates manual optimization. self.automatic_optimization = False - def training_step(batch, batch_idx): + def training_step(self, batch, batch_idx): opt = self.optimizers() opt.zero_grad() loss = self.compute_loss(batch)