728x90 torch seed1 torch seed 고정 코드 딥러닝 학습의 일관성을 위해서 아래와 같이 seed를 고정해주고, 제안된 방법들을 적용해야 효과를 정확히 측정할 수 있습니다. def set_seed_sejun(seed, cuda=True): np.random.seed(seed) torch.manual_seed(seed) random.seed(seed) if cuda: torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False 2023. 7. 15. 이전 1 다음 728x90