Ошибка преобразования TensorRT / Tensorflow2.0 в __saved_model_init_op - PullRequest
Новая
       32

Ошибка преобразования TensorRT / Tensorflow2.0 в __saved_model_init_op

1 голос
/ 06 октября 2019

Я хочу преобразовать свою модель в оптимизированную модель тензоррта для обслуживания тензорного потока, однако я получил следующую ошибку, как я могу это исправить? 

Got a dictionary containing non-Tensor value None for key __saved_model_init_op

Ниже приведены мои сохраненные сигнатуры модели. 

MetaGraphDef with tag-set: 'serve' contains the following SignatureDefs:

signature_def['__saved_model_init_op']:
  The given SavedModel SignatureDef contains the following input(s):
  The given SavedModel SignatureDef contains the following output(s):
    outputs['__saved_model_init_op'] tensor_info:
        dtype: DT_INVALID
        shape: unknown_rank
        name: init_1
  Method name is:

signature_def['serving_default']:
  The given SavedModel SignatureDef contains the following input(s):
    inputs['input_1'] tensor_info:
        dtype: DT_FLOAT
        shape: (-1, 64, -1, 3)
        name: input_1:0
  The given SavedModel SignatureDef contains the following output(s):
    outputs['time_distributed'] tensor_info:
        dtype: DT_FLOAT
        shape: (-1, -1, 7695)
        name: time_distributed/Reshape_1:0
  Method name is: tensorflow/serving/predict

Я думаю, что это проблема на signature_def ['__ved_model_init_op'], которая неизвестна, я сохранил свою модель с ниже 

K.experimental.export_saved_model(model, "{}/01".format(output_path))

Обновление

Это может быть вызваноСлой tf.keras.layers.TimeDistributed, который включен в мою модель.

Это проблем с тензорным потоком в ожидании обновления Tensorflow.

...