[Bug Fix] trainer.update(1) should be used after loss.mean() is called

dmlc · Nov 12, 2019 · e502764 · e502764
1 parent 61ec270
commit e502764
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/sentiment_analysis/sentiment_analysis_cnn.py b/scripts/sentiment_analysis/sentiment_analysis_cnn.py
@@ -159,7 +159,7 @@ def train(net, train_data, test_data, dev_data=None):
                 L = loss(output, label).mean()
             L.backward()
             # Update parameter.
-            trainer.step(args.batch_size)
+            trainer.step(1)
             log_interval_L += L.asscalar()
             epoch_L += L.asscalar()
             if (i + 1) % args.log_interval == 0: