Add manual logging to training_step manual optimization (#4476)

Lightning-AI · Nov 21, 2020 · 69ce966 · 69ce966
1 parent d040946
commit 69ce966
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/optimizers.rst b/docs/source/optimizers.rst
@@ -48,6 +48,10 @@ to manually manage the optimization process. To do so, do the following:
         opt_d.step()
         opt_d.zero_grad()
 
+        # log losses
+        self.log('loss_a', loss_a)
+        self.log('loss_b', loss_b)
+
 .. note:: This is only recommended for experts who need ultimate flexibility
 
 Manual optimization does not yet support accumulated gradients but will be live in 1.1.0