標籤變量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',),否則它假設其他一切都是學習參數，並會嘗試初始化它們，導致出現此錯誤。 Docs：http://mxnet.io/api/python/module.html#mxnet.module.module.Module

所以在我的情況下它需要Module(label_names=('labelidx',), ...)