mxnet ValueError: Unknown initialization pattern for xxx_softmax_label. Default initialization is no

城俊BLOG

于 2020-07-28 16:13:03 发布

阅读量253

点赞数

CC 4.0 BY-SA版权

分类专栏： # DL-报错 MxNet Python

本文链接：https://blog.csdn.net/qxqxqzzz/article/details/107638711

Python 同时被 3 个专栏收录

307 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

DL-报错

122 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

MxNet

43 篇文章

订阅专栏

在使用mxnet构建模型时，由于误将softmax层作为输出层添加导致ValueError。正确做法应为添加softmax激活层而非softmax计算loss的层。参考链接提供了解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码：

layer_out = mx.symbol.Softmax(data=softmax_input, name='gdn_softmax')
# 下面这种也一样
# layer_out = mx.symbol.SoftmaxOutput(data=softmax_input, name='gdn_softmax') # (256,)

报错：

Traceback (most recent call last

了解本专栏

超级会员免费看