机器学习

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.devstorage.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>
上一篇下一篇

猜你喜欢

热点阅读