updated optimizer_step docs

This commit is contained in:
William Falcon 2019-08-13 11:59:33 -04:00
parent fd845d41c0
commit 699fbabda7
1 changed files with 3 additions and 0 deletions

View File

@ -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
```
---