Проблема с предтренированной моделью ALBERT на TF Hub - PullRequest
Новая
       57

Проблема с предтренированной моделью ALBERT на TF Hub

0 голосов
/ 28 октября 2019

Я пытался протестировать модель ALBERT на TF Hub. При попытке загрузить ALBERT из TF Hub я получил следующую ошибку: 

python
Python 3.6.8 (default, May 16 2019, 05:58:38)
[GCC 4.8.5 20150623 (Red Hat 4.8.5-36.0.1)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow_hub as hub
>>> albert_module = hub.Module(
...     "https://tfhub.dev/google/albert_base/1",
...     trainable=True)
Traceback (most recent call last):
  File "<stdin>", line 3, in <module>
  File "/home/opc/vuh/tools/pyvenv3-gpu-tf-hub/lib64/python3.6/site-packages/tensorflow_hub/module.py", line 170, in __init__
    tags=self._tags)
  File "/home/opc/vuh/tools/pyvenv3-gpu-tf-hub/lib64/python3.6/site-packages/tensorflow_hub/native_module.py", line 340, in _create_impl
    name=name)
  File "/home/opc/vuh/tools/pyvenv3-gpu-tf-hub/lib64/python3.6/site-packages/tensorflow_hub/native_module.py", line 382, in __init__
    op.name for op in self._meta_graph.meta_info_def.stripped_op_list.op})
  File "/home/opc/vuh/tools/pyvenv3-gpu-tf-hub/lib64/python3.6/site-packages/tensorflow_hub/native_module.py", line 822, in register_ops_if_needed
    % missing_ops.difference(set(cpp_registry_ops.keys())))
tensorflow.python.framework.errors_impl.NotFoundError: Graph ops missing from the python registry ({'BatchMatMulV2'}) are also absent from the c++ registry.

Кто-нибудь знает почему? Заранее большое спасибо?

К вашему сведению, в моем python3 venv я использую:

наборы данных tenorflow 1.1.0 Оценка-тензор потока 1.13.0 Тензор-gpu 1.13.1 Центр тензор-потока0,6,0 тензорный поток метаданных 0,14,0 тензорный поток вероятности 0,7,0 трф-предложение 0,1,83 предложение 0,1,82

...