Skip to content

Instantly share code, notes, and snippets.

@MFreidank

MFreidank/torch_determinism_flags.py

Created June 27, 2020 09:10
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save MFreidank/d4e62b6ab32e637ec51717f55b633d49 to your computer and use it in GitHub Desktop.
Save MFreidank/d4e62b6ab32e637ec51717f55b633d49 to your computer and use it in GitHub Desktop.
Download ZIP
torch_deterministic_mode
Raw
torch_determinism_flags.py
import torch
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment