Deep learning 모델의 재현성(Reproducibility)을 위해 seed 를 고정할 때 아래 함수를 사용한다.
def seed_everything(seed):
random.seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = True
Python
복사