Pytorch 0.4.0版本保存的模型在高版本调用问题的解决方式
时间:2022-07-26
本文章向大家介绍Pytorch 0.4.0版本保存的模型在高版本调用问题的解决方式,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
在框架升级过程中,经常会出现老版本模型无法调用的问题,其中一个重要的报错经常是:
module.norm1.norm_func.running_mean” and “module.norm1.norm_func.running_var”
for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved
before 0.4.0, this may be expected because InstanceNorm2d does not track running stats
by default since 0.4.0. Please remove these keys from state_dict. If the running stats
are actually needed, instead set track_running_stats=True in InstanceNorm2d to enable
them. See the documentation of InstanceNorm2d for details.
从上面可以看出,模型加载的时候,提醒了老版本的问题。
为了解决这一个问题,可以进行模型中将某些模型进行删除。如下所示:
model_dict = torch.load(args.test_weight_path)
model_dict_clone = model_dict.copy()
for key, value in model_dict_clone.items():
if key.endswith(('running_mean', 'running_var')):
del model_dict[key]
Gnet.load_state_dict(model_dict,False)
而再仔细观察这个问题,发现本质上是一个函数InstanceNorm2d 的关系,因此可以找到该函数,进行修订使其可以支持老版本,即不会出现该问题,解决办法如下:即将track_running_stats=True这个配置新增进去,即不会报错!
norm_layer = functools.partial(
nn.InstanceNorm2d, affine=False, track_running_stats=True)
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- RecyclerView仿应用列表实现网格布局
- Android实现带进度条的WebView
- Android实现记住密码功能
- 【- Flutter Web篇 -】 FlutterUnit web版闪亮登场
- Android简单实现弹幕效果
- 实现 Base64 的编码解码
- Android实现底部导航栏的主界面
- Spring注解@Autowired源码分析
- 使用RecyclerView实现水平列表
- android实现记住用户名和密码以及自动登录
- Android碎片fragment实现静态加载的实例代码
- android异步消息机制 源码层面彻底解析(1)
- /undefinedcss/modules/layer/default/layer.css?v=3.1.1找不到的问题
- Android实现渐变启动页和带有指示器的引导页
- the input device is not a TTY. If you are using mintty, try prefixing the comma