torch seed 값 고정
딥러닝 모델을 개발하다보면, 성능 비교를 위해서는 seed 값 고정이 매우 중요하다.
(seed가 고정되어 있어야 내가 제안한 모델의 성능이 좋은지 안좋은지 확인이 가능하다)
그래서 다음과 같이 고정하는 방법에 대해 정리해 보았다.
import torch
import random
import torch.backends.cudnn as cudnn
torch.manual_seed(0)
torch.cuda.manual_seed(0)
torch.cuda.manual_seed_all(0)
np.random.seed(0)
cudnn.benchmark = False
cudnn.deterministic = True
random.seed(0)
Author And Source
이 문제에 관하여(torch seed 값 고정), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@jaeha0725/torch-seed-값-고정저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)