我正在使用pytorch，不知道是否应该使用torch.backends.cudnn.benchmark = True。我在Google上发现，当计算图不变时，应该使用它。 pytorch中的计算图是什么？
最佳答案
如果您的模型没有更改，并且您的输入大小保持不变-那么您可以从设置torch.backends.cudnn.benchmark = True中受益。
但是，如果您的模型发生了变化:例如，如果您的图层仅在满足某些条件时才被“激活”，或者您的循环中的图层可以重复不同的次数，则设置torch.backends.cudnn.benchmark = True可能会使执行停止。
关于python - 设置 `torch.backends.cudnn.benchmark = True`还是不设置？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/58961768/