pytorch模型运算时间的不同

本文探讨了使用PyTorch框架时遇到的一个现象:相同模型首次调用的计算时间远超过后续调用。作者通过多次实验验证了这一现象,并推测这可能与PyTorch的初始化过程有关。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

今天测试模型计算时间时,运行了这样一段代码:
在这里插入图片描述
在调用时发现 首次调用第一个模型的计算时间 会比之后的模型计算时间长很多,运行结果如图:
在这里插入图片描述

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

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

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值