为什么mxnet.gluon.nn.Dense对象的权重矩阵没有形状?

时间:2017-12-11 15:15:52

标签: mxnet

我尝试遵循这个不错的MXNet Tutorial。我创建了一个非常简单的神经网络(两个输入单元,没有隐藏单元和一个输出单元):

from mxnet import gluon

net = gluon.nn.Dense(1, in_units=2)

之后我尝试看一下权重矩阵的形状(与教程中描述的方式相同):

print(net.weight)

因此我希望看到这个:

Parameter dense4_weight (shape=(1, 2), dtype=None)

但是,我看到以下错误消息:

Traceback (most recent call last):
  File "tmp.py", line 5, in <module>
    print(net.weight)
  File "/usr/local/lib/python3.6/site-packages/mxnet/gluon/parameter.py", line 120, in __repr__
    return s.format(**self.__dict__)
KeyError: 'shape'

我做错了吗?

1 个答案:

答案 0 :(得分:2)

这是一个发生here的回归,并且已在主分支here上修复。预计它将在下一个MXNet版本中修复。