PG1 Baseline - Plant Pathology 2020 - [ 머신러닝-딥러닝 문제해결 전략 ] 12.3.1 시드값 고정 및 GPU 장비 설정 시드값 고정 import torch # 파이토치 import random import numpy as np import os # 시드값 고정 seed = 50 os.environ['PYTHONHASHSEED'] = str(seed) random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False torch.backends.cudnn.enabled = False.. 2023. 10. 2. 이전 1 다음 728x90 반응형
