AI/pytorch
파이토치 환경에서 randomness를 고정하기 위한 방법
교 향
2022. 9. 28. 19:40
자세한 설명은 다른 분들이 적어둔 걸 참고 하고
시간이 없는 나는 일단 적어두고 복붙
def seed_everything(seed=1):
""" Seed and GPU setting """
random.seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True
# torch.backends.cudnn.benchmark = True