AI Hub is deprecated and will no longer be available on Google Cloud after January 23, 2024. All the functionality of AI Hub and new features are available on the Vertex AI platform. See Migrate to Vertex AI to learn how to migrate your resources.
Stay organized with collections
Save and categorize content based on your preferences.
Using a TensorFlow module
This page gives a brief guide to finding a TensorFlow module on
AI Hub and using the module in your machine learning (ML) program.
A TensorFlow module is a self-contained piece of a
TensorFlow graph and the
associated weights and assets. You can use the TensorFlow Hub API
to reuse a module in your TensorFlow program.
By using a TensorFlow module, you can:
Preprocess input feature vectors.
Train a model with a smaller dataset than is necessary if you start from
scratch.
Click TensorFlow module under Category to filter
the list of assets to display only modules, then browse or search to find
a module.
Once you find the TensorFlow module you are interested in,
click the module's name. AI Hub displays a description of
the module and information on how to use it. For more information on how
to find assets, read the guide to finding assets on
AI Hub.
On the module page on AI Hub, click Copy URL under
Use this asset. AI Hub copies
the URL of the TensorFlow module into your clipboard. The URL
is the path to the module directory that you use to instantiate the module in
your code.
The module page on AI Hub includes a code snippet and other
details to help you use the module.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Hard to understand","hardToUnderstand","thumb-down"],["Incorrect information or sample code","incorrectInformationOrSampleCode","thumb-down"],["Missing the information/samples I need","missingTheInformationSamplesINeed","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2025-04-12 UTC."],[[["TensorFlow modules are self-contained parts of a TensorFlow graph, including weights and assets, that can be reused in your machine learning programs via the TensorFlow Hub API."],["Using a TensorFlow module can preprocess input feature vectors, train models with smaller datasets, speed up training, and improve model generalization."],["TensorFlow modules can currently be found on AI Hub by filtering for \"TensorFlow module\" under the \"Category\" section."],["To use a module, you can copy the module's URL from the AI Hub module page, which will then be used to instantiate the module in your code."]]],[]]
