Экспорт модуля тензорного потока - PullRequest
0 голосов
/ 19 декабря 2018

Я хочу сохранить хаб-модуль локально, не создавая системную переменную окружения "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( документ )?Предположительно, если я смог создать спецификацию модуля без него, почему я не могу экспортировать его без него?

Суть вопроса, как я могу кэшировать модуль локально, не прибегая к загрузке его каждый раз из Интернета?

...