Merge pull request #51 from xyzhou-puck/master

fix bugs

examples/bert/bert_classifier.py

@@ -103,7 +103,7 @@ def main():
         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,

examples/bert_leveldb/bert_classifier.py

@@ -105,7 +105,7 @@ def main():
         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,

hapi/text/bert/optimization.py

@@ -130,6 +130,18 @@ class Optimizer(object):
                 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()