美文网首页机器学习
TF Hub 离线使用

TF Hub 离线使用

作者: 孤泉冷月 | 来源:发表于2019-11-25 11:02 被阅读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>
    

    相关文章

      网友评论

        本文标题:TF Hub 离线使用

        本文链接:https://www.haomeiwen.com/subject/kxydwctx.html