非影像相關模型,處理模型時的相關問題

我們是透過Tensorflow字詞轉換成 1x50的向量,訓練DNN神經網路並匯出tflite,現已成功使用官方API將tflite轉換成onnx並優化,面臨以下問題

1.量化階段

我們將(1,30)的input輸入至onnx模型卻顯示輸入錯誤,必須以(1,1,1,30) shape進行輸入,為何與原本模型輸入shape不同

2.NEF階段

我們已將(1,30)透過(1,1,1,30) shape進行輸入,並轉換nef成功,導入耐能加速棒,這時有點疑惑不知怎麼透過加速棒進行推論,我們發現現有的範例都是圖片為主找不到以字向量為主的範本,套用範例程式碼時因圖片的處理方式不同產生與我們當初訓練模型的結果大相逕庭,想請問這方面如何處理呢?

模型結構如上

Comments

The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.