можно ли сделать куда детерминированным? - PullRequest
0 голосов
/ 18 октября 2018

Я пытаюсь провести рефакторинг приложения и проверяю новую версию на старой, сравнивая результаты.На первом слое конвектора pytorch, используя идентичный ввод, я получаю идентичный вывод при использовании CPU.Однако при использовании cuda на GPU выходные тензоры отличаются на 5 десятичных знаков в некоторых ячейках.Есть ли способ устранить эту разницу?Я уже устанавливаю:

random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True

1 Ответ

0 голосов
/ 30 мая 2019

Использование torch.backends.cudnn.benchmark = False

...