This commit is contained in:
thomwolf
2019-08-05 17:16:56 +02:00
parent 70c10caa06
commit 7223886dc9

View File

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