(PyTorch0.4.1) AttributeError: ‘BatchNorm2d‘ object has no attribute ‘track_running_state‘

1. 问题:AttributeError: 'BatchNorm2d' object has no attribute 'track_running_state'

在PyTorch0.4.1环境中运行代码,报错。(代码中使用了预训练模型densenet121、resnet50)(此代码的原作者是在torch==0.2.0.post3,torchvision==0.1.9环境中运行的)

2. 参考:

1)'BatchNorm2d' object has no attribute 'track_running_stats' - #16 by heroinlin - PyTorch Forums

链接中提供的解决方法:将pytorch降级到0.3.1版本。

 

 

 

 2)'BatchNorm2d' object has no attribute 'track_running_stats' - 代码天地

3. 解决:

老版本有‘track_running_stats’这个变量,pytorch0.4.0以后的版本没有这个变量了,相当于是pytorch版本不兼容的问题,所以会出现这个错误。

那么,需要把pytorch降级到pytorch0.3.1版本,即可解决此问题。。。那就重新装一遍pytorch0.3.1呗。。。。

结果发现pytorch0.3.1不支持windows系统(pytorch0.4.0之前,只有linux版本和mac版本,不支持windows。。。),然而我用的是windows系统,所以说,如果想要解决这个问题,需要在linux服务器上重新装一遍pytorch0.3.1环境。。。

如果你也遇到了这个问题且你用的是linux环境,那么直接回退到pytorch0.3.1版本即可。

猜你喜欢

转载自blog.csdn.net/weixin_39450145/article/details/125700431