PyTorch set seed 고정된 검증 결과 얻기
포스트
취소

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

set seed

1
2
3
4
5
6
    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 해서 창을 닫아버리고 다시 열어야 적용이 되는 듯 하다.

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.

Highcharts-jekyll github blog test

matplotlib 한글 나눔폰트