converter = tf.lite.TocoConverter.from_keras_model_file("model2.h5")
tflite_quantized_model=converter.convert()
open("quantized_model.tflite", "wb").write(tflite_quantized_model)
找不到lite,所以tensorflow-gpu=1.8 不行啦
升级到1.13.cudu和cudnn不行
降级到1.12,如下的方案:
https://stackoverflow.com/questions/53914785/how-to-fix-this-module-tensorflow-has-no-attribute-lite
converter = tf.contrib.lite.TFLiteConverter.from_keras_model_file("sample_android.h5")
tflite_model = converter.convert()
open("quantized_model.tflite", "wb").write(tflite_quantized_model)