본문 바로가기
컴퓨터과학

pytorch 코드의 재생산성을 위한 입력코드 정리하는 곳

by 홈카페주인 2023. 6. 11. 
SEED = 1234

random.seed(SEED)
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)
torch.backends.cudnn.deterministic = True

 

 

 

 

참고자료

https://hoya012.github.io/blog/reproducible_pytorch/

 

Reproducible PyTorch를 위한 randomness 올바르게 제어하기!

PyTorch 코드의 완벽한 재현을 위해 고려해야할 randomness를 제어하는 법을 설명드리겠습니다.

hoya012.github.io

https://stackoverflow.com/questions/66130547/what-does-the-difference-between-torch-backends-cudnn-deterministic-true-and

 

What does the difference between 'torch.backends.cudnn.deterministic=True' and 'torch.set_deterministic(True)'?

My network includes 'torch.nn.MaxPool3d' which throw a RuntimeError when cudnn deterministic flag is on according to the PyTorch docs (version 1.7 - https://pytorch.org/docs/stable/generated/torch.

stackoverflow.com

 

728x90

'컴퓨터과학' 카테고리의 다른 글

[트러블슈팅] git repository 특정 폴더가 업로드가 제대로 되지 않고 github에서 화살표로 보이는 현상  (0) 2023.06.30
[트러블슈팅] ModuleNotFoundError: No module named 'torchtext.data.metrics'  (0) 2023.06.12
python import 할때와 pip install 할 때 이름이 서로 다른 패키지 정리하는 페이지  (0) 2023.06.06
FGSM - adversarial training 구현해보기  (0) 2023.06.06
[논문읽기] A combined CNN-LSTM and LSTM-QRNN model for prediction of Idiopathic Pulmonary Fibrosis Progression using CT Scans and Clinical Data  (0) 2023.06.01

관련글

댓글

티스토리툴바