mxnet에 액세스한 gluon 모델 매개변수가 KeyError:'shape'오류 메시지를 표시합니다.

2463 단어 mxnet
코드는 다음과 같습니다.
from mxnet.gluon import nn
from mxnet import nd


def get_net():
    net = nn.Sequential()
    with net.name_scope():
        net.add(nn.Dense(4, activation="relu"))
        net.add(nn.Dense(2))
    return net

x = nd.random.uniform(shape=(3, 5))
net = get_net()
net.initialize()
print(net(x))

w = net[0].weight
b = net[0].bias
print('name: ', net[0].name, '
weight: '
, w, '
bias: '
, b)

오류는 다음과 같습니다.
weight:  Traceback (most recent call last):
  File "E:/python file/distribution_mpc/test1.py", line 22, in 
    print('name: ', net[0].name, '
weight: '
, w, '
bias: '
, b) File "D:\Anaconda\lib\site-packages\mxnet\gluon\parameter.py", line 121, in __repr__ return s.format(**self.__dict__) KeyError: 'shape'

해결 방법: 1. mxnet을 다시 설치합니다. 최신 버전은 이 문제를 개선했습니다.2. 지정된 서류에 가서 한 줄을 고치면 된다.바로 D:\Anaconda\lib\site-packages\mxnet\gluon\parameter.py(구체적인 경로는 자신의 오보 정보를 보아야 한다)의 121행return s.format(**self.__dict__)return s.format(name=self.name, shape=self.shape, dtype=self.dtype)로 바꾸면 됩니다.
여기

좋은 웹페이지 즐겨찾기