Skip to content

Commit 257a351

Browse files
committed
fix pickle dump in run_squad example
1 parent 3951c2c commit 257a351

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/run_squad.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -878,7 +878,7 @@ def main():
878878
if args.local_rank == -1 or torch.distributed.get_rank() == 0:
879879
logger.info(" Saving train features into cached file %s", cached_train_features_file)
880880
with open(cached_train_features_file, "wb") as writer:
881-
train_features = pickle.dump(train_features, writer)
881+
pickle.dump(train_features, writer)
882882
logger.info("***** Running training *****")
883883
logger.info(" Num orig examples = %d", len(train_examples))
884884
logger.info(" Num split examples = %d", len(train_features))

0 commit comments

Comments
 (0)