PyTorch set seed 고정된 검증 결과 얻기

set seed

    def set_seed(self, seed):
        random.seed(seed)
        np.random.seed(seed)
        torch.manual_seed(seed)
        if torch.cuda.is_available():
            torch.cuda.manual_seed_all(seed)

위 함수를 model.train 하기 바로 직전에 걸어주면 되는 듯 하다.

Jupyter notebook 인 경우, 커널 리셋이 아니라 브라우저 자체를 close and halt 해서 창을 닫아버리고 다시 열어야 적용이 되는 듯 하다.