fix #944

2019-08-05 17:16:56 +02:00
parent 70c10caa06
commit 7223886dc9
1 changed files with 1 additions and 0 deletions
--- a/README.md
+++ b/README.md
@@ -385,6 +385,7 @@ for batch in train_data:
    loss.backward()
    scheduler.step()
    optimizer.step()
+    optimizer.zero_grad()
 ```

 ## Citation