torch seed 값 고정

424 단어 PyTorchPyTorch

딥러닝 모델을 개발하다보면, 성능 비교를 위해서는 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)

좋은 웹페이지 즐겨찾기