From 699fbabda7ba809807f85ceeb082fe68b883b19c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 13 Aug 2019 11:59:33 -0400 Subject: [PATCH] updated optimizer_step docs --- docs/Trainer/hooks.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docs/Trainer/hooks.md b/docs/Trainer/hooks.md index 9eb1dbf872..13c58eb3f5 100644 --- a/docs/Trainer/hooks.md +++ b/docs/Trainer/hooks.md @@ -92,6 +92,9 @@ def optimizer_step(self, current_epoch, batch_nb, optimizer, optimizer_i): if batch_nb % 4 == 0 : optimizer.step() optimizer.zero_grad() + + # ... + # add as many optimizers as you want ``` ---