ludwig AttributeError: module 'tensorflow.python.keras.layers.normalization' has no attribute 'BatchNormalizationBase'

raogr8fs  于 2个月前  发布在  Python
关注(0)|答案(3)|浏览(41)

我曾经和Ludwig一起工作,效果很好。我不知道发生了什么,所以我甚至无法导入这个包
from ludwig.api import LudwigModel
它显示了这个错误
AttributeError: module 'tensorflow.python.keras.layers.normalization' has no attribute 'BatchNormalizationBase'

szqfcxe2

szqfcxe21#

嘿,@aminaBm,你使用的TensorFlow和Ludwig的版本是什么?你能提供完整的堆栈跟踪吗?

vmjh9lq9

vmjh9lq92#

抱歉延迟了
tensorflow Version: 1.14.0
ludwig Version: 0.1.0
我进行了pip升级,现在显示
module 'tensorflow.python.framework.ops' has no attribute 'RegisterShape'

zynd9foi

zynd9foi3#

@aminaBm 这个问题在当前的0.5rc2版本中应该不再相关,因为它是基于PyTorch的。你能尝试用当前版本训练相同的模型吗?

相关问题