[그 외] pytorch, random 난수고정
2023. 6. 6. 14:37ㆍAI/그 외
def set_seed(random_seed):
torch.manual_seed(random_seed)
torch.cuda.manual_seed(random_seed)
torch.cuda.manual_seed_all(random_seed) # if use multi-GPU
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
np.random.seed(random_seed)
random.seed(random_seed)
set_seed(42)'AI > 그 외' 카테고리의 다른 글
| [그 외] 허깅페이스에서 repo 내 Files에서 single file 다운 받기 (0) | 2023.05.29 |
|---|---|
| [그 외] huggingface mt5 오류 (0) | 2023.05.19 |
| [그 외] Huggingface Trainer의 Early Stopping (0) | 2023.05.08 |