add random seed for ernie_ie tests (#876)

* add random seed for ernie_ie tests

Author: joey12300

tests/ernie_information_extraction/run_ernie.py

@@ -15,6 +15,8 @@
 from functools import partial
 import argparse
 import os
+import random
+import numpy as np
 
 import paddle
 from paddlenlp.data import Stack, Tuple, Pad
@@ -36,6 +38,13 @@
 # yapf: enable
 
 
+def set_seed(seed):
+    """sets random seed"""
+    random.seed(seed)
+    np.random.seed(seed)
+    paddle.seed(seed)
+
+
 def convert_to_features(example, tokenizer, label_vocab):
     tokens, labels = example
     tokenized_input = tokenizer(
@@ -107,6 +116,7 @@ def create_dataloader(dataset,
     trainer_num = paddle.distributed.get_world_size()
     if trainer_num > 1:
         paddle.distributed.init_parallel_env()
+    set_seed(102)
     # Create dataset, tokenizer and dataloader.
     train_ds, dev_ds, test_ds = load_dataset(
         datafiles=(os.path.join(args.data_dir, 'train.txt'),