This is an automated email from the ASF dual-hosted git repository.
riteshghorse pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new f26afc05e23 [Python] Add note for using TF2 models with TensorFlow Hub
(#27276)
f26afc05e23 is described below
commit f26afc05e2387cf3050b1ac92a209d12e5d3baa7
Author: Ritesh Ghorse <[email protected]>
AuthorDate: Wed Jun 28 12:44:57 2023 -0400
[Python] Add note for using TF2 models with TensorFlow Hub (#27276)
* add note for tf2 models
* Update examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
Co-authored-by: Rebecca Szper <[email protected]>
* Update examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
Co-authored-by: Rebecca Szper <[email protected]>
---------
Co-authored-by: Rebecca Szper <[email protected]>
---
.../beam-ml/run_inference_with_tensorflow_hub.ipynb | 19 +++++++++++--------
1 file changed, 11 insertions(+), 8 deletions(-)
diff --git a/examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
b/examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
index 63f1f7ccd02..08a6d3e64b4 100644
--- a/examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
+++ b/examples/notebooks/beam-ml/run_inference_with_tensorflow_hub.ipynb
@@ -40,7 +40,7 @@
"metadata": {
"id": "fFjof1NgAJwu"
},
- "execution_count": 1,
+ "execution_count": null,
"outputs": []
},
{
@@ -104,7 +104,10 @@
"cell_type": "markdown",
"source": [
"## Use TensorFlow Hub's trained model URL\n",
- "To use TensorFlow Hub's trained model URL, pass the model URL to the
`model_uri` field of `TFModelHandler` class."
+ "To use TensorFlow Hub's trained model URL, pass the model URL to the
`model_uri` field of `TFModelHandler` class.\n",
+ "\n",
+ "**Note:** Only models that you save in the [TF2
format](https://www.tensorflow.org/tutorials/keras/save_and_load#save_the_entire_model)
are compatible with `TFModelHandler`.\n",
+ "To see TF2 models, go to the [TF2 section of the TensorFlow
Hub](https://tfhub.dev/s?subtype=module,placeholder&tf-version=tf2)."
],
"metadata": {
"id": "dVLTtFxDuJT_"
@@ -120,7 +123,7 @@
"metadata": {
"id": "H4-ZvkcTv7MO"
},
- "execution_count": 3,
+ "execution_count": null,
"outputs": []
},
{
@@ -132,7 +135,7 @@
"metadata": {
"id": "n3M6FNaUwBbl"
},
- "execution_count": 4,
+ "execution_count": null,
"outputs": []
},
{
@@ -154,7 +157,7 @@
"id": "q23ip_HkwL3G",
"outputId": "051bfa77-ce1f-4ee2-abcd-26ae7f037180"
},
- "execution_count": 5,
+ "execution_count": null,
"outputs": [
{
"output_type": "stream",
@@ -187,7 +190,7 @@
"metadata": {
"id": "QLFCEisBwPlz"
},
- "execution_count": 6,
+ "execution_count": null,
"outputs": []
},
{
@@ -230,7 +233,7 @@
"id": "MSKMc_s6wSUH",
"outputId": "73695bae-75ae-4f02-a0a5-750531b8f90b"
},
- "execution_count": 8,
+ "execution_count": null,
"outputs": [
{
"output_type": "stream",
@@ -249,4 +252,4 @@
]
}
]
-}
+}
\ No newline at end of file
