当使用keras训练,后台为tensorflow时报错TypeError: softmax() got an unexpected keyword argument ‘axis’ 解决办法
此种方法可能为在Dense(4, activation=‘softmax’)(x) 时调用了tf.nn.softmax(x,axis=axis)中的axis,可以进入tensorflow_backend.py中把tf.nn.softmax(x,axis=axis)改为tf.nn.softmax(x,dim=axis),可以避免冲突,从而运行程序。网上也有另外一种说法是tensorfloe版本太低,进行升级至高版本也可以解决(还没有尝试)。
TypeError: softmax() got an unexpected keyword argument 'axis' 解决办法
最新推荐文章于 2024-06-12 21:01:46 发布