把softmax换成sigmoid之后就运行报错
收藏
在飞机识别任务中https://aistudio.baidu.com/aistudio/projectdetail/169475
运行环境是aistudio
二分类问题想把项目中的softmax函数
#predict = fluid.layers.fc(h2,size=1,act='sigmoid')
predict = fluid.layers.fc(h2,size=2,act='softmax')
换成sigmoid函数后
predict = fluid.layers.fc(h2,size=1,act='sigmoid')
训练就报错了
请问该如何解决呢?
0
收藏
请登录后评论
把size改成2就行,既然是二分类结果肯定有两种