import numpy import torch import random def set_rng_seeds(seed): random.seed(seed) numpy.random.seed(seed) torch.random.manual_seed(seed) torch.cuda.manual_seed_all(seed) # print('RNG_SEED {}'.format(seed))
一款轻量级的自然语言处理(NLP)工具包,目标是减少用户项目中的工程型代码,例如数据处理循环、训练循环、多卡运行等