必做题:
(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设备;
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)