'BatchNorm2d' object has no attribute 'track_running

'BatchNorm2d' object has no attribute 'track_running_stats'

还不知道什么原因：

后来发现是老版本训练的权重，用0.4.0以后版本打开，这个变量没有，不能兼容。

我的解决方法：

load模型后，报存pth，再新建网络，加载模型，理论上能用了。

import pickle
pickle.load = partial(pickle.load, encoding="latin1")
pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1")
self.net = torch.load(graph_path, map_location=lambda storage, loc: storage, pickle_module=pickle)try:torch.save(self.net.state_dict(), '111.pth')
except Exception as e:print(e)torch.save(self.net.module.features.state_dict(), '111.pth')
# self.net = torch.load(graph_path, map_location=lambda storage, loc: storage)
self.net.eval()

用pytorch加载训练好的模型的时候遇到了如下的问题：

AttributeError: 'module' object has no attribute '_

'BatchNorm2d' object has no attribute 'track_running_stats'相关推荐

yolov5 v3.0训练报错： torch.nn.modules.module.ModuleAttributeError: ‘BatchNorm2d‘ object has no attribute
欢迎大家关注笔者,你的关注是我持续更博的最大动力原创文章,转载告知,盗版必究 yolov5 v3.0版本训练报错:torch.nn.modules.module.ModuleAttributeErr ...
'ScriptModel' object has no attribute 'save'
'ScriptModel' object has no attribute 'save' 原因:torch0.4.0没有保存功能 import torchclass ScriptModel(torch ...
Python错误：AttributeError: 'generator' object has no attribute 'next'解决办法
今天在学习生成器对象(generation object)运行以下代码时,遇到了一个错误: #定义生成器函数 def liebiao(): for x in range(10): yield x #函 ...
AttributeError: 'dict' object has no attribute 'status_code'
前端AJAX请求数据,提示错误:"AttributeError: 'dict' object has no attribute 'status_code'". 原因:是提示返回对象 ...
Traceback (most recent call last): File AttributeError: 'NoneType' object has no attribute 'group'
Traceback (most recent call last):File "<stdin>", line 1, in <module> Attribut ...
解决：AttributeError: ‘Graph‘ object has no attribute ‘number_of_selfloops‘
解决:AttributeError: 'Graph' object has no attribute 'number_of_selfloops' 目录解决:AttributeError: 'Grap ...
AttributeError: ‘FPDF‘ object has no attribute ‘unifontsubset‘
AttributeError: 'FPDF' object has no attribute 'unifontsubset' 目录 AttributeError: 'FPDF' object has ...
AttributeError: ‘Series‘ object has no attribute ‘as_matrix‘
AttributeError: 'Series' object has no attribute 'as_matrix' 问题: y_test = test_shifted["y_t+1&q ...
AttributeError: ‘SVC‘ object has no attribute ‘_probA‘
AttributeError: 'SVC' object has no attribute '_probA' 问题: # Save the Modle to file in the current w ...

'BatchNorm2d' object has no attribute 'track_running_stats'

'BatchNorm2d' object has no attribute 'track_running_stats'相关推荐

最新文章

热门文章