pytorch
2a585173 - Set torch.backends.cudnn.enabled to false when testing accuracy (#94363)

Commit
1 year ago
Set torch.backends.cudnn.enabled to false when testing accuracy (#94363) Summary: It looks like setting torch.backends.cudnn.deterministic to True is not enough for eliminating non-determinism when testing benchmarks with --accuracy, so let's turn off cudnn completely. With this change, mobilenet_v3_large does not show random failure on my local environment. Also take this chance to clean up CI skip lists. Pull Request resolved: https://github.com/pytorch/pytorch/pull/94363 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading