Source code for gperc.utils
import torch
import numpy as np
import random
[docs]def set_seed(s=4):
"""set seed for controlling randomness"""
random.seed(s)
np.random.seed(s)
torch.manual_seed(s)
torch.cuda.manual_seed(s)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False