《Pytorch 模型推理及多任务通用范式》第三节课作业

《Pytorch 模型推理及多任务通用范式》第三节课作业,第1张

《Pytorch 模型推理及多任务通用范式》第三节课作业

必做题:

(1)把模型改为resnet18,加载相应的模型权重,跑一下0.jpg 和 1.jpg,看一下输出结果。

          0.jpg        'umbrella', 0.9995712637901306

          1.jpg        'peacock', 0.9999839067459106

(2)自己找2张其他图,用resnet18做下推理。

         

          'goldfish', 0.9763382077217102

         

        'bald eagle', 0.9960240125656128         

思考题: 

 (1)以ResNet18为例,用time模块和for循环,对0.jpg连续推理100 次,统计时间开销

          在CPU上,推理100次“0.jpg” 共耗时9.184秒;

          无GPU设备;

 

 

 

 

 

 

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/4655489.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-06
下一篇 2022-11-06

发表评论

登录后才能评论

评论列表(0条)

保存