中午在工位睡不着,闲着无聊用keras构建了一个四个卷积层,输出是10个类别的卷积神经网络,
BUT 问题来了,最后一个输出层的激活函数是softmax。报了softmax的axis=axis参数没有,显示是后端tensorflow的问题,第一感觉是keras版本和后台tensorflow版本不适配。上网查了查,确实因为版本问题,
两种办法,一个是更改keras版本
一个是tensorflow报错文件修改axis=axis ====>dim=axis