[转载]torch参数:torch.backends.cudnn.benchmark的意义

[转载]torch参数:torch.backends.cudnn.benchmark的意义

来源：https://zhuanlan.zhihu.com/p/73711222

完整版请看原文，这里只截取一部分结论性的东西：

设置 torch.backends.cudnn.benchmark=True 将会让程序在开始时花费一点额外时间，为整个网络的每个卷积层搜索最适合它的卷积实现算法，进而实现网络的加速。适用场景是网络结构固定（不是动态变化的），网络的输入形状（包括 batch size，图片大小，输入的通道）是不变的，其实也就是一般情况下都比较适用。反之，如果卷积层的设置一直变化，将会导致程序不停地做优化，反而会耗费更多的时间。