2019年8月13日
torch.backends.cudnn.benchmark = true 使用情形
摘要： 设置这个 flag 可以让内置的 cuDNN 的 auto-tuner 自动寻找最适合当前配置的高效算法，来达到优化运行效率的问题。 应该遵循以下准则： 如果网络的输入数据维度或类型上变化不大，设置 torch.backends.cudnn.benchmark = true 可以增加运行效率； 如果 阅读全文
posted @ 2019-08-13 17:52 U_C 阅读(927) 评论(0) 推荐(0) 编辑