将这个flag设置为TRUE的话,每次返回的卷积算法是确定的,即默认算法。
如果结合前置torch中设置随机数seed一定的话,可以保证每次运行网络相同输入得到的输出是相同的。
torch.backends.cudnn.deterministic
最新推荐文章于 2024-08-15 16:43:28 发布
将这个flag设置为TRUE的话,每次返回的卷积算法是确定的,即默认算法。
如果结合前置torch中设置随机数seed一定的话,可以保证每次运行网络相同输入得到的输出是相同的。