Edge TPU Compiler: ERROR: quantized_dimension must be in range [0, 1). Was 3
问题 I'm trying to get a Mobilenetv2 model (retrained last layers to my data) to run on the Google edge TPU Coral. I've followed this tuturial https://www.tensorflow.org/lite/performance/post_training_quantization?hl=en to do the post-training quantization. The relevant code is: ... train = tf.convert_to_tensor(np.array(train, dtype='float32')) my_ds = tf.data.Dataset.from_tensor_slices(train).batch(1) # POST TRAINING QUANTIZATION def representative_dataset_gen(): for input_value in my_ds.take(30)