Pytorch 시드 고정하는 법

실험의 재현 가능성을 확보하기 위해 시드를 고정해야 한다. 

 

파이토치 시드 고정

 

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) # gpu 1개 이상일 때 

# 넘파이 랜덤시드 고정
np.random.seed(0)

#CuDNN 랜덤시드 고정
cudnn.benchmark = False
cudnn.deterministic = True # 연산 처리 속도가 줄어들어서 연구 후반기에 사용하자

# 파이썬 랜덤시드 고정
random.seed(0)