torch.backends.cudnn.benchmark 和torch.backends.cudnn.deterministic
torch.backends.cudnn.benchmark标志位True or False cuDNN是GPU加速库 在使用GPU的时候,PyTorch会默认使用cuDNN加速,但是,在使用 cuDNN 的时候,torch.backends.cudnn.benchmark模式是为False。 设置这个 flag 为True,我们就可以在 PyTorch 中对模型里的卷积层进行预先的优化,也就是在…