torch调用模型推理结果不对分析

Love The Way You Lie 2022-10-30 15:30 118阅读 0赞

此处修改后可用代码：[https://github.com/BIGBALLON/CIFAR-ZOO/blob/master/eval.py][https_github.com_BIGBALLON_CIFAR-ZOO_blob_master_eval.py]

# 1.模型是否导入成功 #

## 1.1 问题 ##

Missing key(s) in state_dict: "features.0.weight"
    Unexpected key(s) in state_dict: "module.features.0.weight"

例如，如果此时使用的下面命令导入模型：

net.load_state_dict(checkpoint["state_dict"], strict=False)

此处strict参数设置为False（会忽略不匹配的网络层参数）。如何训练时使用的多GPU并行训练的模型名称中，会多"module."，模型可能会导入不成功。

参考：

[https://blog.csdn.net/shiwanghualuo/article/details/101756876][https_blog.csdn.net_shiwanghualuo_article_details_101756876]

[https://blog.csdn.net/qq\_32998593/article/details/89343507][https_blog.csdn.net_qq_32998593_article_details_89343507]

## 1.2 解决 ##

net.load_state_dict({k.replace('module.',''):v for k,v in checkpoint["state_dict"].items()}, strict=True)

# 2.输入图片预处理 #

torch常常会使用PIL读取图片，transforms里面的一些预处理（resize、ToTensor等等）。之前验证过PIL与opencv只需要都转为相同的颜色空间，就基本没问题了。

[https_github.com_BIGBALLON_CIFAR-ZOO_blob_master_eval.py]: https://github.com/BIGBALLON/CIFAR-ZOO/blob/master/eval.py
[https_blog.csdn.net_shiwanghualuo_article_details_101756876]: https://blog.csdn.net/shiwanghualuo/article/details/101756876
[https_blog.csdn.net_qq_32998593_article_details_89343507]: https://blog.csdn.net/qq_32998593/article/details/89343507

发表评论取消回复

表情：

评论列表（有 0 条评论，118人围观）

还没有评论，来说两句吧...

相关阅读

相关 vLLM模型推理引擎参数大全

llm ai对话大语言模型推理服务框架引擎参数大全

╰半夏微凉°/ 2024年04月22日 15:28/ 0 赞/ 60 阅读

相关 PyTorch Hub发布！一行代码调用所有模型：torch.hub

Facebook官方博客表示，PyTorch Hub是一个简易API和工作流程，为复现研究提供了基本构建模块，包含预训练模型库。并且，PyTorch Hub还支持Col...

红太狼/ 2024年04月20日 08:45/ 0 赞/ 50 阅读

相关 torch 对参数处理后回填

先从网络中下载参数，将参数所有元素整合在一个向量中，对每个元素添加一定的噪声，接下来还原向量中的元素到参数的对应位置，并用加噪后的参数回填到模型中，最后检查一遍回填情况。假

谁践踏了优雅/ 2023年10月14日 16:43/ 0 赞/ 32 阅读

相关 Bert中文分类模型：训练+推理+部署

BERT（Bidirectional Encoder Representation from Transformers）是google-research在2018年10月提出的

骑猪看日落/ 2023年09月28日 08:15/ 0 赞/ 120 阅读

相关 mysql自定义结果_MySQL自定义函数调用不出结果

自定义函数的代码： DROP FUNCTION IF EXISTS fn\_HrStaffBase\_GetNameFromidCard delimiter // CRE

亦凉/ 2023年01月12日 07:50/ 0 赞/ 125 阅读

相关 torch调用模型推理结果不对分析

此处修改后可用代码：[https://github.com/BIGBALLON/CIFAR-ZOO/blob/master/eval.py][https_github.com_

Love The Way You Lie/ 2022年10月30日 15:30/ 0 赞/ 119 阅读

相关深度学习模型大小与模型推理速度的探讨

导读 ![1deab23f895f63a95f0805a04828c2a9.gif][] 作者：田子宸，毕业于浙江大学，就职于商汤，文章经过作者同意转载。本文将对

悠悠/ 2022年08月28日 08:49/ 0 赞/ 202 阅读

相关行测-图形推理题分析

笔试中做了一些图形推理题，总结如下，规律也都是在不断的摸索中总结出来的 1.特殊情况（1）有阴影的要想到阴影部分的面积和旋转的情况 ![在这里插入图片描述][70

不念不忘少年蓝@/ 2022年05月09日 10:58/ 0 赞/ 253 阅读

相关 Java调用Tensorflow训练模型预测结果

Java调用Tensorflow训练好的模型做预测，首先需要读取词典，然后加载模型，读入数据，最后预测结果。模型训练参考上一篇博客：[使用Tensorflow训练LSTM

╰半夏微凉°/ 2022年04月10日 05:54/ 0 赞/ 333 阅读

相关人工智能--不确定性推理概述

文章目录不确定性推理的含义什么是不确定性推理为什么采用不确定性推理不确定性推理的基本问题不确定性的表示

超、凢脫俗/ 2022年01月21日 02:15/ 0 赞/ 1006 阅读