2021年11月11日
cudnn.benchmark = True
摘要： 在很多情况下我们能看到代码有这样一行： 1 torch.backends.cudnn.benchmark = true 而且大家都说这样可以增加程序的运行效果，那到底有没有这样的效果，或者什么情况下应该这样做呢？ 解决办法： 总的来说，大部分情况下，设置这个flag可以让内置的cuDNN的auto- 阅读全文
posted @ 2021-11-11 11:06 重大的小鸿 阅读(916) 评论(0) 推荐(0)