Update train_eval.py

fix a bug

Author: 649453932

train_eval.py

@@ -6,7 +6,7 @@
 from sklearn import metrics
 import time
 from utils import get_time_dif
-from pytorch_pretrained_bert.optimization import BertAdam
+from pytorch_pretrained.optimization import BertAdam
 
 
 # 权重初始化，默认xavier
@@ -118,4 +118,4 @@ def evaluate(config, model, data_iter, test=False):
         report = metrics.classification_report(labels_all, predict_all, target_names=config.class_list, digits=4)
         confusion = metrics.confusion_matrix(labels_all, predict_all)
         return acc, loss_total / len(data_iter), report, confusion
-    return acc, loss_total / len(data_iter)
+    return acc, loss_total / len(data_iter)