Skip to content

Commit 9238d56

Browse files
committed
Update benchmark_utils.py
1 parent 9bbc581 commit 9238d56

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/benchmarks/benchmark_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -782,7 +782,7 @@ def train_step():
782782
def train_step():
783783
with mxnet.autograd.record():
784784
contextual_embedding, pooled_out = model(input_ids, valid_length,
785-
input_ids, valid_length)
785+
input_ids, valid_length)
786786
fake_loss = (contextual_embedding * contextual_embedding_ograd).sum() \
787787
+ (pooled_out * pooled_out_ograd).sum()
788788
fake_loss.backward()

0 commit comments

Comments
 (0)