keras构建模型的插曲

    xiaoxiao2022-07-04  130

    中午在工位睡不着,闲着无聊用keras构建了一个四个卷积层,输出是10个类别的卷积神经网络,

    BUT 问题来了,最后一个输出层的激活函数是softmax。报了softmax的axis=axis参数没有,显示是后端tensorflow的问题,第一感觉是keras版本和后台tensorflow版本不适配。上网查了查,确实因为版本问题,

    两种办法,一个是更改keras版本

    一个是tensorflow报错文件修改axis=axis   ====>dim=axis

    最新回复(0)