如何从本地系统加载 TF hub 模型 [英] How to load TF hub model from local system

问题描述

一种方法是每次从 tensorflow_hub 下载模型，如下所示

One way is to download the model each time from tensorflow_hub like following

import tensorflow as tf
import tensorflow_hub as hub

hub_url = "https://tfhub.dev/google/tf2-preview/nnlm-en-dim128/1"
embed = hub.KerasLayer(hub_url)
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)

我想下载一次文件并反复使用，而不必每次都下载

I want to download the file once and use again and again with out downloading each time

推荐答案

1. 从 url + ?tf-hub-format=compressed"下载您的模型
   例如https://tfhub.dev/google/tf2-preview/nnlm-en-dim128/1?tf-hub-format=compressed"
2. 解压
3. 在代码中加载解压后的文件夹
   

import tensorflow as tf
import tensorflow_hub as hub

embed = hub.KerasLayer('path/to/untarred/folder')
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)

这篇关于如何从本地系统加载 TF hub 模型的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！