PyTorch에서 Random seed 고정하는 방법입니다. 저도 까먹고 있다가 맨날 찾고 싶은데 찾기 귀찮아서 공유해둡니다. 여러가지 방법이 있겠지만, 모르겠다 싶을 때는 랜덤적인 요소들은 모두 찾아서 고정해버리는게 방법이죠. import os import random import numpy as np import torch def set_random_seed(seed=42): os.environ["PYTHONHASHSEED"] = str(seed) random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) torch.backends.cud..