torch.backends.cudnn.deterministic

将这个flag设置为TRUE的话,每次返回的卷积算法是确定的,即默认算法。
如果结合前置torch中设置随机数seed一定的话,可以保证每次运行网络相同输入得到的输出是相同的。

你可能感兴趣的:(torch.backends.cudnn.deterministic)