From 9710b68dbcdfc2b94264a68d35885ee70faaeb2f Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Tue, 18 Jun 2019 16:01:15 +0200
Subject: [PATCH] fix pickles

---
 examples/run_classifier.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/examples/run_classifier.py b/examples/run_classifier.py
index 20bc59ded8..45131103ab 100644
--- a/examples/run_classifier.py
+++ b/examples/run_classifier.py
@@ -40,6 +40,12 @@ from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule
 
 from run_classifier_dataset_utils import processors, output_modes, convert_examples_to_features, compute_metrics
 
+if sys.version_info[0] == 2:
+    import cPickle as pickle
+else:
+    import pickle
+
+
 logger = logging.getLogger(__name__)