From c588453a0fc8469b96b0a055bd301516f0a33261 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 30 Nov 2018 14:22:40 +0100 Subject: [PATCH] fix run_squad --- examples/run_squad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_squad.py b/examples/run_squad.py index e3213189bf..041a51a879 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -878,7 +878,7 @@ def main(): if args.local_rank == -1 or torch.distributed.get_rank() == 0: logger.info(" Saving train features into cached file %s", cached_train_features_file) with open(cached_train_features_file, "wb") as writer: - train_features = pickle.dump(train_features, writer) + pickle.dump(train_features, writer) logger.info("***** Running training *****") logger.info(" Num orig examples = %d", len(train_examples)) logger.info(" Num split examples = %d", len(train_features))