将模型转换为 TFLite 格式会得到"Check failed: 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。这在我们的路线图上，这些文档描述了现有的操作支持。