Python 如何在自定义路径中保存/加载tensorflow hub模块？

tensorflow\u hub

库维护人员使用户可以轻松下载和使用预先培训的tensorflow模块，例如：

import tensorflow_hub as hub

embed = hub.Module("https://tfhub.dev/google/universal-sentence-encoder/1")

但是从

sys.stderr

来看，它似乎是在本地将模块保存到一个临时目录，即

信息：tensorflow：使用 /var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_模块到 缓存模块。信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片0:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_0信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片1:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_1信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片10:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带有嵌入项/分片信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片11:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_11信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片12:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_12信息：tensorflow:初始化变量 来自检查点的模块/嵌入\u en/分片\u 13:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带有嵌入项/分片信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片14:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_14信息：tensorflow:初始化变量 来自检查点的模块/嵌入\u en/分片\u 15:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带有嵌入项/分片信息：tensorflow:初始化变量 来自检查点的模块/嵌入\u en/分片\u 16:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_16信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片2:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_2信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片3:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_3信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片4:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_4信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片5:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_5信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片6:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_6信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片7:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_7信息：tensorflow:初始化变量 来自检查点的模块/嵌入\u en/分片\u 8:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_8信息：tensorflow:初始化变量 来自检查点的模块/嵌入/分片9:0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带嵌入的_en/shared_9信息：tensorflow:初始化变量 模块/编码器\u en/DNN/ResidualHidden\u 0/权重：检查点0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带编码器_en/DNN/ResidualHidden_0/weights 信息：tensorflow:初始化变量 模块/编码器\u en/DNN/ResidualHidden\u 1/权重：检查点0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带编码器_en/DNN/ResidualHidden_1/weights 信息：tensorflow:初始化变量 模块/编码器\u en/DNN/ResidualHidden\u 2/权重：检查点0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'，带编码器_en/DNN/ResidualHidden_2/权重 信息：tensorflow:初始化变量 模块/编码器\u en/DNN/ResidualHidden\u 3/投影：来自检查点的0 b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_modules/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/variables/variables'带编码器_en/DNN/ResidualHidden_3/投影 信息：tensorflow:初始化变量 模块/编码器\u en/DNN/ResidualHidden\u 3/权重：检查点0 带编码器的b'/var/folders/j6/xczfl75n3sbfwpg4190gpb104vnlxt/T/tfhub_模块/c6f5954ffa065cdb2f2e604e740e8838bf21a2d3/变量/变量'

cache_dir = os.path.join(tempfile.gettempdir(), "tfhub_modules")

import tensorflow_hub as hub 
embed = hub.load('/home/admin/embed/')

os.environ["TFHUB_CACHE_DIR"] = '/tmp/tfhub'