Merge pull request PaddlePaddle#51 from xyzhou-puck/master

fix bugs
qingqing01 · Apr 21, 2020 · 4e92b27 · 4e92b27
2 parents 7dcbcc9 + 14c2d97
commit 4e92b27
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/examples/bert/bert_classifier.py b/examples/bert/bert_classifier.py
@@ -103,7 +103,7 @@ def mnli_line_processor(line_id, line):
         batch_size=config.batch_size,
         line_processor=mnli_line_processor)
 
-    dev_dataloader = BertDataLoader(
+    test_dataloader = BertDataLoader(
         "./data/glue_data/MNLI/dev_matched.tsv",
         tokenizer, ["contradiction", "entailment", "neutral"],
         max_seq_length=config.max_seq_len,

diff --git a/examples/bert_leveldb/bert_classifier.py b/examples/bert_leveldb/bert_classifier.py
@@ -105,7 +105,7 @@ def mnli_line_processor(line_id, line):
         mode="leveldb",
         phase="train")
 
-    dev_dataloader = BertDataLoader(
+    test_dataloader = BertDataLoader(
         "./data/glue_data/MNLI/dev_matched.tsv",
         tokenizer, ["contradiction", "entailment", "neutral"],
         max_seq_length=config.max_seq_len,

diff --git a/hapi/text/bert/optimization.py b/hapi/text/bert/optimization.py
@@ -130,6 +130,18 @@ def exclude_from_weight_decay(self, name):
                 return True
         return False
 
+    def state_dict(self):
+        return self.optimizer.state_dict()
+
+    def set_dict(self, state_dict):
+        return self.optimizer.set_dict(state_dict)
+
+    def get_opti_var_name_list(self):
+        return self.optimizer.get_opti_var_name_list()
+
+    def current_step_lr(self):
+        return self.optimizer.current_step_lr()
+
     def minimize(self, loss, use_data_parallel=False, model=None):
         param_list = dict()