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