mxnet運行時遇到問題及解決方法


1.訓練好模型之后,進行預測時出現這種錯誤:

1 mxnet.base.MXNetError: [15:05:50] src/ndarray/ndarray.cc:239: Check failed: from.shape() == to->shape() operands shape mismatchfrom.shape = (63,) to.shape=(11,)

這種問題的解決方法,在全鏈接層fc中設置num_hidden你需要分類的個數和訓練的模型的實際分類的個數參數不匹配,查看模型json,修改為對應值即可,這里修改成63及63個分類即可!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM