Tensorflow Hub: застрял при импорте модели - PullRequest
Новая
       28

Tensorflow Hub: застрял при импорте модели

0 голосов
/ 23 октября 2018

Попытка импортировать некоторые модели с помощью Tensorflow Hub с этим кодом: 

import tensorflow as tf
import tensorflow_hub as hub

elmo_model = hub.Module('https://tfhub.dev/google/elmo/2', trainable=True)

Застревает мой ноутбук.Единственная строка журнала, появляющаяся перед застреванием:

INFO: tenorflow: Использование / tmp / tfhub_modules для кэширования модулей.

Как открепить его и разрешить импортмодели от Tensorflow Hub?

1 Ответ

0 голосов
/ 23 октября 2018

Это было просто из-за привилегий: я не смог получить доступ к каталогу по умолчанию, где Tensorflow Hub хранит модели (/tmp/tfhub_modules).

Чтобы решить эту проблему, я просто выбираю каталог для хранения моделей, которые яможно получить доступ: 

import os
import tensorflow as tf
import tensorflow_hub as hub

os.environ['TFHUB_CACHE_DIR'] = '/home/user/workspace/tf_cache'
elmo_model = hub.Module('https://tfhub.dev/google/elmo/2', trainable=True)
...