Как преобразовать более одного файла .pth в один файл .pb? - PullRequest
Новая
       87

Как преобразовать более одного файла .pth в один файл .pb?

0 голосов
/ 04 августа 2020

Если бы кто-нибудь мог мне помочь, я был бы очень признателен!

Я хотел бы спросить, как насчет преобразования из файла .pth в .pb.

В частности, я использую MADDPG алгоритм с pytorch, и у меня есть 4 .pth файлов .

Как я могу преобразовать их все и получить один файл .pb?

Во-первых, Я попытался преобразовать один из моих 4 файлов .pth в файл .pb с помощью этих инструкций

Как мы можем преобразовать модель .pth в файл .pb?

но Я получаю эту ошибку. 

C:\Python36\lib\site-packages\onnx_tf\common\__init__.py:89: UserWarning: 
onnx_tf.common.get_outputs_names is deprecated. It will be removed in future release. Use 
TensorflowGraph.get_outputs_names instead.
  warnings.warn(message)
Traceback (most recent call last):
  File "myscript.py", line 2, in <module>
    from onnx_tf.backend import prepare
  File "C:\Python36\lib\site-packages\onnx_tf\__init__.py", line 1, in <module>
    from . import backend
  File "C:\Python36\lib\site-packages\onnx_tf\backend.py", line 29, in <module>
   from onnx_tf.common.handler_helper import get_all_backend_handlers
  File "C:\Python36\lib\site-packages\onnx_tf\common\handler_helper.py", line 5, in <module>
    from onnx_tf.handlers.backend import *  # noqa
  File "C:\Python36\lib\site-packages\onnx_tf\handlers\backend\average_pool.py", line 7, in <module>
from .pool_mixin import PoolMixin
  File "C:\Python36\lib\site-packages\onnx_tf\handlers\backend\pool_mixin.py", line 12, in <module>
from .dilated_pooling import DilatedPooling
  File "C:\Python36\lib\site-packages\onnx_tf\handlers\backend\dilated_pooling.py", line 6, in 
 <module>
from onnx_tf.common import pooling_helper
  File "C:\Python36\lib\site-packages\onnx_tf\common\pooling_helper.py", line 16, in <module>
pad_tf_ops = pad_ops(tf.maximum, tf.math.ceil, tf.math.floor,

AttributeError: module 'tensorflow' has no attribute 'math'

1 Ответ

0 голосов
/ 04 августа 2020

Я использовал 1.7.0. Я обновляю его до 2.30, но теперь получаю следующую ошибку: 

 File "C:\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 64, 
in <module>
 from tensorflow.python._pywrap_tensorflow_internal import *
ImportError: DLL load failed: The specified module could not be found.

During handling of the above exception, another exception occurred:

   Traceback (most recent call last):
     File "myscript.py", line 2, in <module>
    from onnx_tf.backend import prepare
    File "C:\Python36\lib\site-packages\onnx_tf\__init__.py", line 1, in <module>
    from . import backend
    File "C:\Python36\lib\site-packages\onnx_tf\backend.py", line 21, in <module>
    import tensorflow as tf
    File "C:\Python36\lib\site-packages\tensorflow\__init__.py", line 41, in <module>
    from tensorflow.python.tools import module_util as _module_util
    File "C:\Python36\lib\site-packages\tensorflow\python\__init__.py", line 40, in 
    <module>
    from tensorflow.python.eager import context
    File "C:\Python36\lib\site-packages\tensorflow\python\eager\context.py", line 35, 
in 
    <module>
    from tensorflow.python import pywrap_tfe
    File "C:\Python36\lib\site-packages\tensorflow\python\pywrap_tfe.py", line 28, in 
    <module>
    from tensorflow.python import pywrap_tensorflow
    File "C:\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 
   83, 
    in <module>
    raise ImportError(msg)
    ImportError: Traceback (most recent call last):
    File "C:\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 
64, 
   in <module>
    from tensorflow.python._pywrap_tensorflow_internal import *
   ImportError: DLL load failed: The specified module could not be found.


   Failed to load the native TensorFlow runtime.

See https://www.tensorflow.org/install/errors

for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.
...