今天测试模型计算时间时,运行了这样一段代码:
在调用时发现 首次调用第一个模型的计算时间 会比之后的模型计算时间长很多,运行结果如图:
可以看到即便是同样是resnet50,第一次调用和第二次调用时间差别非常大。而且即便我改变了模型加载的顺序,在调用时仍然是第一个模型的计算时间远超其他模型。
因此我判断python程序在调用pytorch框架下第一个模型计算时,运算时间会比之后再计算时间长很多,并且用循环体做了另一个实验,我运行了这样一段程序:
可以看到即使是同样的模型,第一次调用它计算的所耗的时间也比之后长的多,这样的现象应该和pytorch的初始化有关。