Skip to content
Snippets Groups Projects
loaders.py 218 B
Newer Older
Alice Brenon's avatar
Alice Brenon committed
import numpy
import random
import torch

def set_random():
    seed_value = 42
    random.seed(seed_value)
    numpy.random.seed(seed_value)
    torch.manual_seed(seed_value)
    torch.cuda.manual_seed_all(seed_value)