加入tf.contrib.quantize.create_train_graph()和tf.contrib.quantize.create_eval_graph()两个伪量化函数在训练代码中,属于完全量化的一个步骤,就是quantization aware training 方法,加入两句话是实现伪量化训练过程,自动添加伪量化节点,找到最大最小值,这样生成的pb文件是含有最大最小值的,模拟了伪量化过程,是完全量化的第一步。第二步骤则需要采用 Pb2tflite代码 进行转化成tflite模型。
待续。。。。