亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

嘗試使用張量流 2 在已保存的模型上進行預測時出錯

嘗試使用張量流 2 在已保存的模型上進行預測時出錯

慕娘9325324 2022-09-06 16:39:07
我正在嘗試使用以下代碼對保存的模型進行預測 features = np.ones((20, 40, 3), dtype=np.float32) features = tf.convert_to_tensor(value, dtype=tf.float32) imported_model = tf.saved_model.load(export_dir=os.path.join(os.path.join(model_path, directory))) import_fn = imported_model.signatures["serving_default"] import_fn(features)使用Tensorflow 2運行時,我收到以下錯誤。當我使用saved_model_cli時,模型預測工作正常。tensorflow.python.framework.errors_impl.InvalidArgumentError:  In[0] is not a matrix. Instead it has shape [20,40,3]     [[node dense/BiasAdd (defined at model_manager.py:54) ]] [Op:__inference_pruned_318590]保存的 cli 命令如下所示saved_model_cli run --dir ./model_dir --tag_set serve --signature_def serving_default --input_exprs 'input=np.ones((20, 40, 3), dtype=np.float32)'
查看完整描述

1 回答

?
慕少森

TA貢獻2019條經驗 獲得超9個贊

無效參數錯誤通常是由輸入中的數據類型不匹配引起的。

基于你的錯誤“In[0]不是一個矩陣。相反,它具有形狀[20,40,3]”。
您可以嘗試操作輸入數據,以正確匹配最初訓練模型的輸入類型和形狀。您還可以檢查與 Python IDE 相比,使用saved_model_cli時模型如何處理您的輸入。由于您在使用 Python IDE 時可能會錯過一些預處理步驟,而在使用saved_model_cli。

您可以在此鏈接中閱讀有關使用Saved_Model格式用法的更多信息


查看完整回答
反對 回復 2022-09-06
  • 1 回答
  • 0 關注
  • 71 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號