Skip to content

GitLab

Commit 7223886d authored by thomwolf's avatar thomwolf
Browse files

fix #944

parent 70c10caa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ 7223886d
...@@ -385,6 +385,7 @@ for batch in train_data: ...@@ -385,6 +385,7 @@ for batch in train_data:
loss.backward() loss.backward()
scheduler.step() scheduler.step()
optimizer.step() optimizer.step()
optimizer.zero_grad()
``` ```
## Citation ## Citation
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment