Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-09-10 10:56:56 +02:00
parent d1b6929043
commit 8fe2bf1d7f
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/torch_utils.py
View File

@ -6,7 +6,11 @@ def init_seeds(seed=0):
torch.manual_seed(seed) torch.manual_seed(seed)
torch.cuda.manual_seed(seed) torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed) torch.cuda.manual_seed_all(seed)
# torch.backends.cudnn.deterministic = True # https://pytorch.org/docs/stable/notes/randomness.html
# Remove randomness (may be slower on Tesla GPUs) # https://pytorch.org/docs/stable/notes/randomness.html
if seed == 0:
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
def select_device(force_cpu=False, apex=False): def select_device(force_cpu=False, apex=False):