mt-dnn icon indicating copy to clipboard operation
mt-dnn copied to clipboard

Reproducibility issue with CuDNN backend

Open songyouwei opened this issue 5 years ago • 2 comments

https://github.com/namisan/mt-dnn/blob/25c3bd1cfee2db14c7e8dbec4a369acaa95b15f6/data_utils/utils.py#L25

When running on the CuDNN backend, there might be reproducibility issue. According to pytorch document, two further options must be set:

torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

songyouwei avatar May 19 '19 03:05 songyouwei

Thanks pointing it out!! Does this work on pytorch version 0.4x? We will test it soon.

namisan avatar May 19 '19 07:05 namisan

I haven't tested it on pytorch version 0.4x.

songyouwei avatar May 19 '19 08:05 songyouwei