将模型转换为 TFLite 格式得到“检查失败：is_rnn_state_array”

Question

我想转成 TFLite 格式

toco \--input_file=$TRAINING_DIR/retrained_graph.pb \
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE \
--output_file=/$TRAINING_DIR/${ARCHITECTURE}.tflite \
--inference_type=QUANTIZED_UINT8 \
--input_arrays=input \
--output_arrays=final_result \ 
--input_shapes=1,224,224,3 \inference_input_type=QUANTIZED_UNIT8 \
--mean_values=128 \
--std_values=128 \
--default_ranges_min=0 \
--quantize_weights=true \
--default_ranges_max=6 

失败了

 F tensorflow/contrib/lite/toco/graph_transformations/quantize.cc:600] Check failed: is_rnn_state_array 

我错过了什么？

Answer 1

TOCO 目前不支持将 RNN 和 LSTM 转换为 TFLite。这是我们的路线图，这些docs 描述了现有的操作支持。