首页 > 其他 > 详细

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

时间:2019-11-29 16:29:55      阅读:79      评论:0      收藏:0      [点我收藏+]

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

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

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

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

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

原文:https://www.cnblogs.com/jiading/p/11958569.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!