This commit is contained in:
thomwolf 2019-08-05 17:16:56 +02:00
parent 70c10caa06
commit 7223886dc9
1 changed files with 1 additions and 0 deletions

View File

@ -385,6 +385,7 @@ for batch in train_data:
loss.backward()
scheduler.step()
optimizer.step()
optimizer.zero_grad()
```
## Citation