pytorch模型运算时间的不同

今天测试模型计算时间时,运行了这样一段代码:
pytorch模型运算时间的不同_第1张图片
在调用时发现 首次调用第一个模型的计算时间 会比之后的模型计算时间长很多,运行结果如图:
pytorch模型运算时间的不同_第2张图片

可以看到即便是同样是resnet50,第一次调用和第二次调用时间差别非常大。而且即便我改变了模型加载的顺序,在调用时仍然是第一个模型的计算时间远超其他模型。

因此我判断python程序在调用pytorch框架下第一个模型计算时,运算时间会比之后再计算时间长很多,并且用循环体做了另一个实验,我运行了这样一段程序:
pytorch模型运算时间的不同_第3张图片
可以看到即使是同样的模型,第一次调用它计算的所耗的时间也比之后长的多,这样的现象应该和pytorch的初始化有关。
pytorch模型运算时间的不同_第4张图片

你可能感兴趣的:(python,人工智能,pytorch)