mirror of https://github.com/tensorflow/hub.git
13 lines
649 B
Markdown
13 lines
649 B
Markdown
WARNING: This code is deprecated.
|
|
|
|
The `retrain.py` tool from this directory has been replaced by the
|
|
[`make_image_classifier`](https://www.tensorflow.org/hub/tutorials/tf2_image_retraining)
|
|
tool that gets
|
|
[installed](https://www.tensorflow.org/hub/installation) as a command-line tool
|
|
by the PIP package `tensorflow-hub[make_image_classifier]`.
|
|
The new tool uses TensorFlow 2 and supports fine-tuning.
|
|
|
|
The Colab notebook
|
|
[tf2_image_retraining.ipynb](https://colab.research.google.com/github/tensorflow/docs/blob/master/site/en/hub/tutorials/tf2_image_retraining.ipynb)
|
|
explains the basic technique behind that tool: transfer learning with TF Hub.
|