首页 Paddle框架 帖子详情
把softmax换成sigmoid之后就运行报错
收藏
快速回复
Paddle框架 问答深度学习 1353 1
把softmax换成sigmoid之后就运行报错
收藏
快速回复
Paddle框架 问答深度学习 1353 1

在飞机识别任务中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
收藏
回复
全部评论(1)
时间顺序
thinc
#2 回复于2020-04

把size改成2就行,既然是二分类结果肯定有两种

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户