CNN 使用交叉熵报错
收藏
使用交叉熵的时候报错了,孩子不知道怎么解决了
0
收藏
请登录后评论
请检查一下你输入的predits和labels的维度,
如果是要做6分类,那你网络的输出层维度也就是predits的维度最好也是6。
而且类别标签编号也不要超过6