作用:加速网络训练
具体实现:为训练前的网络模型的每个卷积层找到最优实现方法,进而实现网络加速
要求:网络模型以及输入数据的维度在训练过程中不变化(一般情况均符合)
torch.backends.cudnn.benckmark = True
最新推荐文章于 2024-05-13 14:42:18 发布
作用:加速网络训练
具体实现:为训练前的网络模型的每个卷积层找到最优实现方法,进而实现网络加速
要求:网络模型以及输入数据的维度在训练过程中不变化(一般情况均符合)