Skip to content

Latest commit

 

History

History
12 lines (10 loc) · 649 Bytes

README.md

File metadata and controls

12 lines (10 loc) · 649 Bytes

WARNING: This code is deprecated.

The retrain.py tool from this directory has been replaced by the make_image_classifier tool that gets installed 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 explains the basic technique behind that tool: transfer learning with TF Hub.