Fix hyp dataset loading (#1900)

Author: joey12300

examples/language_model/ernie-doc/run_classifier.py

@@ -165,7 +165,7 @@ def do_train(args):
         if eval_name == test_name:
             test_ds = eval_ds
         else:
-            test_ds = load_dataset(args.dataset, splits=["train", test_name])
+            test_ds = load_dataset(args.dataset, splits=[test_name])
 
     num_classes = len(train_ds.label_list)
 

paddlenlp/transformers/ernie_doc/modeling.py

@@ -280,7 +280,7 @@ def _cache_mem(self, curr_out, prev_mem):
         if self.mem_len is None or self.mem_len == 0:
             return None
         if prev_mem is None:
-            new_mem = curr[:, -self.mem_len:, :]
+            new_mem = curr_out[:, -self.mem_len:, :]
         else:
             new_mem = paddle.concat([prev_mem, curr_out],
                                     1)[:, -self.mem_len:, :]