TF Hub 离线使用
2019-11-25 本文已影响0人
孤泉冷月
Ubuntu 18.04 LTS
python3.7
tensorflow2
tensorflow_hub 0.7.0
以tf2-preview/gnews-swivel-20dim
为例
https://tfhub.dev/google/tf2-preview/gnews-swivel-20dim/1
对应下载连接为
https://storage.googleapis.com/tfhub-modules/google/tf2-preview/gnews-swivel-20dim/1.tar.gz
即替换tfhub.dev
为storage.googleapis.com/tfhub-modules
,并且在末尾加上后缀.tar.gz
然后将文件解压到自己喜欢的路径
例如
我将它解压到/home/tf_hub/embeding/
/home/tf_hub/embeding/embeding
├── assets
│ └── tokens.txt
├── saved_model.pb
└── variables
├── variables.data-00000-of-00001
└── variables.index
使用方式
使用本地的URL即可
import tensorflow_hub as hub
# embedding = "https://tfhub.dev/google/tf2-preview/gnews-swivel-20dim/1"
embedding = "/home/tf_hub/embeding/"
hub_layer = hub.KerasLayer(embedding,
input_shape=[],
dtype=tf.string,trainable=True)
print(hub_layer)
# <tensorflow_hub.keras_layer.KerasLayer at 0x7f69785de8d0>