1 课程学习
本节课主要对于大白AI课程https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg
《Pytorch模型推理及范式》课程中第三节课进行学习
2 作业解答
必做题
- 把模型改为 resnet18,加载相应的模型权重(Lesson2 的物料包中有),跑一下 0.jpg
和 1.jpg,看一下输出结果。官方 torchvision 训练 mobilenet 和训练 resnet 的方式是一样
的,所以数据预处理和数据后处理部分完全相同
(‘umbrella’, 0.9998866319656372)
(‘peacock’, 0.9999724626541138)
-
自己找 2 张其他图,用 resnet18 做下推理。
(‘Persian cat’, 0.2359200119972229)
(‘airliner’, 0.6159605383872986)
思考题
-
以 ResNet18 为例,用 time 模块和 for 循环,对”./images/0.jpg”连续推理 100
次,统计时间开销
3.6195597648620605
(没有GPU)