我想知道有沒有什么辦法可以看到tensorflow中模型的引數。pytorch 中有一個命令，即 torch.load('/filepath')。
uj5u.com熱心網友回復：
假設您已經將模型保存在MODEL_PATH，這應該可以解決問題：
model = tf.keras.models.load_model(MODEL_PATH)
model.summary()
查看此以獲取有關保存和加載模型的更多資訊。
uj5u.com熱心網友回復：
對于預測背景關系，您可以執行
model = tf.keras.models.load_model(PATH, compile=True)
這適用于 .h5 keras 模型和 SavedModel 格式的模型。否則，您可能必須提供在預測背景關系中可能沒有的自定義指標和訓練代碼。
如需參考，請在此處查看：https : //www.tensorflow.org/api_docs/python/tf/keras/models/load_model
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/351047.html