c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix #944

Browse Source
This commit is contained in:
thomwolf committed 2019-08-05 17:16:56 +02:00
1 parent 70c10caa06
commit 7223886dc9
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1
View File
@@ -385,6 +385,7 @@ for batch in train_data:
loss.backward()
scheduler.step()
optimizer.step()
optimizer.zero_grad()
```
## Citation