torch.backends.cudnn.benchmark

1 作用

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

2 用法

import torch
torch.backends.cudnn.benchmark = True

你可能感兴趣的:(Pytorch,中的各种函数,pytorch)