mxnet infer_shape_entry TypeError: требуется целое число

Question

Я столкнулся с ошибкой infer_shape при тестировании моего пользовательского оператора.

Сообщение об ошибке:

Error in <my_operator_name>.infer_shape: Traceback (most recent call last):
  File "~/anaconda2/envs/mx110-py27/lib/python2.7/site-packages/mxnet/operator.py", line 658, in infer_shape_entry
    array('I', rshape[i])),
TypeError: an integer is required

Кто-нибудь сталкивался с такой же проблемой раньше?Спасибо!

Answer 1

Я обнаружил, что это потому, что функция __init__ в mxnet.operator.CustomProp принимает только аргументы строкового типа.