提交 5c0b25f8 编写于 作者: G guosheng

Change random to np.random to be consistent between py2 and py3 in Transformer

上级 504c7f15
import glob
import os
import random
import tarfile
import numpy as np
class SortType(object):
GLOBAL = 'global'
......@@ -203,7 +204,8 @@ class DataReader(object):
self._token_delimiter = token_delimiter
self.load_src_trg_ids(end_mark, fpattern, start_mark, tar_fname,
unk_mark)
self._random = random.Random(x=seed)
self._random = np.random
self._random.seed(seed)
def load_src_trg_ids(self, end_mark, fpattern, start_mark, tar_fname,
unk_mark):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册