교 향
Maximalism
교 향
전체 방문자
오늘
어제
  • 전체글
    • Coding
    • Linux
    • Docker
    • Python
    • AI
      • GPT
      • Pandas
      • Numpy
      • tensorflow
      • pytorch
      • openCV
      • Study
      • AIconnect - [GPT]한국어..
    • Data structure, OS
    • Competitions
    • Investment
    • Book
    • Health
    • Trip
    • Memo
    • 대학원

블로그 메뉴

  • 홈
  • 태그
  • 미디어로그
  • 위치로그
  • 방명록

공지사항

인기 글

태그

  • ActivationFunction #활성화함수 #인공지능면접
  • AI면접 #딥러닝면접 #인공지능면접 #기울기소실 #GradientVanishing

최근 댓글

최근 글

티스토리

hELLO · Designed By 정상우.
교 향

Maximalism

AI/pytorch

파이토치 환경에서 randomness를 고정하기 위한 방법

2022. 9. 28. 19:40
https://tootouch.github.io/experiment/reproduction_pytorch/
https://hoya012.github.io/blog/reproducible_pytorch/

자세한 설명은 다른 분들이 적어둔 걸 참고 하고

시간이 없는 나는 일단 적어두고 복붙

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

'AI > pytorch' 카테고리의 다른 글

copy.deepcopy(model.state_dict())  (0) 2022.09.28
    'AI/pytorch' 카테고리의 다른 글
    • copy.deepcopy(model.state_dict())
    교 향
    교 향
    AI/ML/DL

    티스토리툴바