[Pytorch] Seed 고정

 

Pytorch 학습 시, seed를 고정하는 방법은 아래와 같다.

로스가 항상 같이 나오니 비교하기 좋다.

import random

SEED = 10

random.seed(SEED)
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)
torch.backends.cudnn.deterministic = True