TensorFlow 模型导出

在這裡的例子的話,對於使用繼承 tf.keras.Model 類別建立的 Keras 模型,是把 call 方法以 @tf.function 修飾,從而轉化為 SavedModel 支持的計算圖。如果你另外建立了一個 predict 方法並且也用 @tf.function 修飾轉換成計算圖,當然也可以調用 model.predict

1 Like