如何在下载TensorFlow Hub模块时避免生成带哈希的自动目录
如何避免TF Hub下载模型时生成哈希目录
用户运行以下代码时,TF Hub会在指定的TFHUB_CACHE_DIR路径下自动生成哈希子目录:
import os path = "https://tfhub.dev/google/universal-sentence-encoder-multilingual/3" + "?tf-hub-format=compressed" os.environ["TFHUB_CACHE_DIR"] = '/use_multilingual/20220406235959/' hub.load(path)
生成的路径示例:
/use_multilingual/20220406235959/328a8dfead197f0fe7268e59acba163b3a015b15
可以通过手动下载并解压模型的方式避免生成哈希目录,具体步骤如下:
- 手动下载模型压缩包:直接访问链接
https://tfhub.dev/google/universal-sentence-encoder-multilingual/3?tf-hub-format=compressed,将压缩包下载到本地。 - 解压压缩包到目标路径:把压缩包内的所有文件直接解压到你指定的
/use_multilingual/20220406235959/路径下,确保解压后该路径下直接是模型的配置文件、权重文件等,没有额外的子目录。 - 修改代码加载本地模型:无需设置
TFHUB_CACHE_DIR(或保留设置也不影响),直接加载本地路径即可:
import hub # 直接加载本地路径的模型 model = hub.load('/use_multilingual/20220406235959/')
补充说明:TF Hub默认生成哈希目录是为了实现缓存机制,不同的模型或版本会对应不同的哈希值,避免同名模型之间的冲突。如果不需要缓存复用,手动下载解压的方式就能满足直接将模型存到指定路径的需求。
内容的提问来源于stack exchange,提问作者madst
相关产品推荐
相关产品推荐

