2019年11月26日
Pytorch学习0.01:cudnn.benchmark= True的设置
摘要： 设置这个 flag 可以让内置的 cuDNN 的 auto tuner 自动寻找最适合当前配置的高效算法，来达到优化运行效率的问题。 应该遵循以下准则： 如果网络的输入数据维度或类型上变化不大，也就是每次训练的图像尺寸都是一样的时候，设置 torch.backends.cudnn.benchmark 阅读全文
posted @ 2019-11-26 22:18 dgi 阅读(5734) 评论(1) 推荐(2) 编辑