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

1 课程学习

本节课主要对于大白AI课程https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg

《Pytorch模型推理及范式》课程中第三节课进行学习

2 作业解答

必做题

1. 把模型改为 resnet18，加载相应的模型权重（Lesson2 的物料包中有），跑一下 0.jpg

和 1.jpg，看一下输出结果。官方 torchvision 训练 mobilenet 和训练 resnet 的方式是一样

的，所以数据预处理和数据后处理部分完全相同

(‘umbrella’, 0.9998866319656372)
(‘peacock’, 0.9999724626541138)

2. 自己找 2 张其他图，用 resnet18 做下推理。
   
   

   (‘Persian cat’, 0.2359200119972229)
   (‘airliner’, 0.6159605383872986)

思考题

1. 以 ResNet18 为例，用 time 模块和 for 循环，对”./images/0.jpg”连续推理 100

   次，统计时间开销

   3.6195597648620605

   （没有GPU）

标签：resnet18,模型,jpg,Pytorch,课程,三节课,推理,多任务
来源： https://blog.csdn.net/weixin_41727238/article/details/120856215