1 作用
对于网络中输入的数据维度和类型相差不大是, 可以使用, 将其设置为 True 可以使计算机自动寻找最高效的算法来进行计算, 从而节省时间, 提高运行效率, 而每次输入变化较大时, 则不适用, 反而会因为频繁的寻找算法而增加程序的运行时间。
2 用法
import torch
torch.backends.cudnn.benchmark = True
对于网络中输入的数据维度和类型相差不大是, 可以使用, 将其设置为 True 可以使计算机自动寻找最高效的算法来进行计算, 从而节省时间, 提高运行效率, 而每次输入变化较大时, 则不适用, 反而会因为频繁的寻找算法而增加程序的运行时间。
import torch
torch.backends.cudnn.benchmark = True