본문 바로가기

Programming

모든 랜덤값을 고정하기 위한 함수 (python)

728x90
반응형

경진 대회든 실제 업무에서의 연구든 재현성을 가지는 것은 중요하다. 특히 gpu를 사용하는 경우 재현성을 확보하기 어려워지는 경우도 있다. 

재현성을 확보하기 위해 다음과 같은 함수를 사용할 수도 있다. 

def seed_everything(seed=1234):
	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
728x90
반응형