如何使用 Tensorflow Hub 调优学习模型?


TensorFlow Hub 是一个包含已训练机器学习模型的存储库。这些模型已准备好随时随地进行微调和部署。诸如 BERT 和 Faster R-CNN 等已训练模型可以通过几行代码重复使用。它是一个开放存储库,这意味着任何人都可以使用和修改它。

该 tfhub.dev 存储库包含许多经过预训练的模型。其中一些包括文本嵌入、图像分类模型、TF.js/TFLite 模型等。

可以使用以下代码进行安装

!pip install --upgrade tensorflow_hub

可以将其导入工作环境,如下面的代码所示

import tensorflow_hub as hub

更新于: 2021-2-25

246 次浏览

开始你的 职业

通过完成课程获得认证

开始了解
广告