导出一个Tensorflow Hub模块

时间:2018-12-19 08:33:16

标签: tensorflow save export tensorflow-hub

我想在本地保存集线器模块,而不创建系统环境变量“ TFHUB_CACHE_DIR”。 我想到的两种方式都需要我没有或不知道如何生产的参数。

选项1

module_url = "https://..."
module_file_name = "path/to/a/module_dir"

session = ???

module_instance = hub.Module(module_url)
module_instance.export(module_file_name, session)

为什么我需要一个会话对象,所以保存模块(doc)?如果我没有模块就可以创建模块,为什么我不能在没有模块的情况下导出模块呢?

选项2。

module_url = "https://..."
module_file_name = "path/to/a/module_dir"
module_spec = hub.load_module_spec(module_url)
module_instance = hub.Module(module_spec)

checkpoint_path = ???

module_spec.export(module_file_name, checkpoint_path)

为什么我需要一个checkpoint_path对象(doc)?如果我没有它就可以创建模块规范,为什么我不能在没有它的情况下导出它呢?

最重要的问题是,如何才能在本地缓存模块,而不必每次都从网络上下载它?

0 个答案:

没有答案