Skip to content

GitLab

Commit 9710b68d authored by thomwolf's avatar thomwolf
Browse files

fix pickles

parent 15ebd67d
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
examples/run_classifier.py
View file @ 9710b68d
...@@ -40,6 +40,12 @@ from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule ...@@ -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 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__) logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment