hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 4e59d88724
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '4e59d88724'
${ noResults }
fastNLP/reproduction/text_classification/utils/util_init.py

12 lines
229 B
Raw Blame History 

  1. import numpy

  2. import torch

  3. import random

  4. 

  5. 

  6. def set_rng_seeds(seed):

  7.     random.seed(seed)

  8.     numpy.random.seed(seed)

  9.     torch.random.manual_seed(seed)

  10.     torch.cuda.manual_seed_all(seed)

  11.     # print('RNG_SEED {}'.format(seed))