如何使用 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 次瀏覽

開始你的 職業

透過完成課程獲得認證

開始瞭解
廣告
© . All rights reserved.