Spaces:
Runtime error
Runtime error
import numpy as np | |
import os | |
import random | |
import torch | |
def seed_all(seed): | |
np.random.seed(seed) | |
os.environ['PYTHONHASHSEED'] = str(seed) | |
random.seed(seed) | |
torch.manual_seed(seed) | |
torch.cuda.manual_seed(seed) | |
torch.backends.cudnn.deterministic = True |