Update benchmark_utils.py

liuzh47 · Aug 11, 2020 · 9238d56 · 9238d56
1 parent 9bbc581
commit 9238d56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/benchmarks/benchmark_utils.py b/scripts/benchmarks/benchmark_utils.py
@@ -782,7 +782,7 @@ def train_step():
             def train_step():
                 with mxnet.autograd.record():
                     contextual_embedding, pooled_out = model(input_ids, valid_length,
-                                                            input_ids, valid_length)
+                                                             input_ids, valid_length)
                     fake_loss = (contextual_embedding * contextual_embedding_ograd).sum() \
                                 + (pooled_out * pooled_out_ograd).sum()
                     fake_loss.backward()