From 53d0abc88f2540453698138bad826429a53e9697 Mon Sep 17 00:00:00 2001 From: Jensun Ravichandran Date: Tue, 18 May 2021 20:29:36 +0200 Subject: [PATCH] Fix incorrect code-snippet in optimizers doc `training_step(...)` should take `self` as the first argument. It's a simple but necessary fix. --- docs/source/common/optimizers.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)