《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)

  1. 自己找 2 张其他图,用 resnet18 做下推理。
    《Pytorch模型推理及多任务通用范式》_第三节课
    《Pytorch模型推理及多任务通用范式》_第三节课

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

思考题

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

    次,统计时间开销

    3.6195597648620605

    (没有GPU)

上一篇:从零开始的思维训练


下一篇:python一键将所在文件夹下的所有jpg转为ico图标