Skip to content

Commit

Permalink
Switch tf2_image_retraining from "tf2-preview" to permanent ("imagene…
Browse files Browse the repository at this point in the history
…t") URLs.

PiperOrigin-RevId: 275447634
  • Loading branch information
TensorFlow Hub Authors authored and andresusanopinto committed Oct 18, 2019
1 parent 84ac11a commit bdc213e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions examples/colab/tf2_image_retraining.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -128,9 +128,9 @@
"id": "mmaHHH7Pvmth"
},
"source": [
"## Select the Hub/TF2 module to use\n",
"## Select the TF2 SavedModel module to use\n",
"\n",
"Hub modules for TF 1.x won't work here, please use one of the selections provided."
"hub.Modules for TF 1.x won't work here."
]
},
{
Expand All @@ -143,9 +143,9 @@
},
"outputs": [],
"source": [
"module_selection = (\"mobilenet_v2\", 224) #@param [\"(\\\"mobilenet_v2\\\", 224)\", \"(\\\"inception_v3\\\", 299)\"] {type:\"raw\", allow-input: true}\n",
"module_selection = (\"mobilenet_v2_100_224\", 224) #@param [\"(\\\"mobilenet_v2_100_224\\\", 224)\", \"(\\\"inception_v3\\\", 299)\"] {type:\"raw\", allow-input: true}\n",
"handle_base, pixels = module_selection\n",
"MODULE_HANDLE =\"https://tfhub.dev/google/tf2-preview/{}/feature_vector/4\".format(handle_base)\n",
"MODULE_HANDLE =\"https://tfhub.dev/google/imagenet/{}/feature_vector/4\".format(handle_base)\n",
"IMAGE_SIZE = (pixels, pixels)\n",
"print(\"Using {} with input size {}\".format(MODULE_HANDLE, IMAGE_SIZE))\n",
"\n",
Expand Down

0 comments on commit bdc213e

Please sign in to comment.