Ошибка инициализации MXNet в переменной метки

Question

Когда я звоню module.fit() Я получаю сообщение об ошибке ValueError: Unknown initialization pattern for labelidx. Символ «labelidx» - это имя, которое я использую для данных моих ярлыков. Я не хотел использовать softmax_label, потому что я не использую выход softmax, но это кажется по умолчанию для многих тиков. Кажется, он пытается инициализировать labelidx в качестве параметра, что является ошибкой. Как я могу сказать, что это вход, а не научный параметр?

Answer 1

Я понял это.

При создании объекта Module необходимо указать ему имена данных (data_names) и наклейки (label_names). Каждый из них должен быть списком имен строк. По умолчанию data_names=('data',), label_names=('softmax_label',), В противном случае предполагается, что все остальное будет изучено и попытается инициализировать их, что приведет к этой ошибке. Документы: http://mxnet.io/api/python/module.html#mxnet.module.module.Module

Так что в моем случае это необходимо Module(label_names=('labelidx',), ...)