torch.no_grad()与.forward()在预测时的性能差异
背景使用pytorch在模型做推断时,使用torch.no_grad()与.forward()都可以得到正确的预测结果,如下:我好奇想知道这两种推断方式那种,那种效率更高,于是随手做个测试。测试输入一张图片,然后推断10000次,看下两种方式各用多少时间:torch.no_grad():.forward:结论我原本以为这两种方式推断效率应该是差不多的,但测试结果告诉我,使用torch.no_grad()这种方式要比.forward()这种方式效率高不少,而且处理任务越多,
原创
2020-11-07 16:48:30 ·
661 阅读 ·
0 评论