дамп ядра mxnet при попытке простой программы

mxnet 1.6.0 выдает дамп ядра, когда я пробую следующую программу

from mxnet import np, autograd
x = np.arange(4.0)
x.attach_grad()
print(x.grad)

with autograd.record():
    y = 2 * np.dot(x, x)
print("y = ", y)
y.backward()

print("x.grad ", x.grad)

mxnet coredumps в autograd.record (): строка ...

terminate вызывается после выдачи экземпляра 'dmlc :: Error' what (): [14:42:13] src / imperative /./ imperative_utils.h: 146: Оператор _npi_multiply_scalar при выводе форм завершился неудачно. формы ввода: нет формы вывода: нет атрибуты оператора: скаляр: 2,0

Трассировка стека: [bt] (0) /home/nisar/anaconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x307d3b) [0x7f90eb02fd3b] [bt] (1) / home / nisar / anaconda3 /lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::imperative::SetShapeType(mxnet::Context const &, nnvm :: NodeAttrs const &, std :: vector ‹mxnet :: NDArray *, std: : allocatormxnet :: NDArray * ›const &, std :: vector‹ mxnet :: NDArray *, std :: allocatormxnet :: NDArray * ›const &, mxnet :: DispatchMode *) + 0x363b) [0x7f90ee1cb3bb]

Пожалуйста помоги


python mxnet autograd
person nak    schedule 15.07.2020    source источник
comment
предоставьте дополнительную информацию об отладке (например, сообщения об ошибках)   -  person Jan    schedule 15.07.2020
comment
Спасибо, Ян, за ответ. Я только начал использовать mxnet, и нам нужно использовать mxnet numpy. Подробности смотрите в ответе.   -  person nak    schedule 15.07.2020

Ответы (1)


arrow_upward
0
arrow_downward

Я сослался на https://github.com/apache/incubator-mxnet/issues/15766 и получил ответ. Мы должны использовать mxnet numpy. Добавлена ​​строка mxnet.npx.set_np ()

Итак, я изменил код на:

from mxnet import np, autograd
mxnet.npx.set_np()
x = np.arange(4.0)
x.attach_grad()
print(x.grad)

with autograd.record():
    y = 2 * np.dot(x, x)
print("y = ", y)
y.backward()

print("x.grad ", x.grad)

#print("x.grad == 4 * x : ", x.grad == 4 * x)
with autograd.record():
    y = x.sum()
y.backward()
print("x.grad = ", x.grad)

и он отлично работает

person nak    schedule 15.07.2020