diff --git a/train_nsf_sim_cache_sid_load_pretrain.py b/train_nsf_sim_cache_sid_load_pretrain.py index 5833dc7..ef8effd 100644 --- a/train_nsf_sim_cache_sid_load_pretrain.py +++ b/train_nsf_sim_cache_sid_load_pretrain.py @@ -9,6 +9,7 @@ n_gpus = len(hps.gpus.split("-")) from random import shuffle, randint import torch + torch.backends.cudnn.deterministic = False torch.backends.cudnn.benchmark = False from torch.nn import functional as F