728x90
반응형
경진 대회든 실제 업무에서의 연구든 재현성을 가지는 것은 중요하다. 특히 gpu를 사용하는 경우 재현성을 확보하기 어려워지는 경우도 있다.
재현성을 확보하기 위해 다음과 같은 함수를 사용할 수도 있다.
def seed_everything(seed=1234):
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
728x90
반응형
'Programming' 카테고리의 다른 글
[monitoring tools] wandb (weights & biases) (0) | 2022.12.06 |
---|---|
[monitoring tools] Tensorboard (0) | 2022.12.05 |
[pytorch] 전이 학습, Transfer Learning (0) | 2022.12.03 |
[pytorch] 학습 결과 및 모델 저장하기 (0) | 2022.12.03 |
[pytorch] Dataset & Dataloader (0) | 2022.12.03 |