torch.manual_seed(seed)
np.random.seed(seed)
random.seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False'Code > Pytorch' 카테고리의 다른 글
| [Pytorch, timm] Optimizer & Parameter Group Learning Rate (0) | 2023.06.27 |
|---|---|
| [Pytorch] Batch 개수, Data 개수 (0) | 2023.05.14 |
| Pytorch GPU에 맞는 CUDA version 설치 (0) | 2023.04.14 |
| [Torch] pytorch version 확인 (0) | 2023.04.14 |
| [Pytorch] Tensor 에서 특정 index 의 값을 뽑아 새로운 tensor 정의 (0) | 2023.04.12 |