恢复ckpt后如何量化（tensorflow lite）

Question

这就是我保存模型的方式

checkpoint_path= "./train" 
ckpt = tf.train.Checkpoint(encoder=encoder,
                                   decoder=decoder,
                                   optimizer = optimizer) 
ckpt_manager = tf.train.CheckpointManager(ckpt, checkpoint_path, max_to_keep=5)

在训练时我在每个 epoch 后保存

ckpt_manager.save()

后来恢复模型

ckpt = tf.train.Checkpoint(encoder=encoder,decoder=decoder,optimizer = optimizer)
ckpt.restore(checkpoint_path)

这些是检查点路径中的文件

现在，这是我的问题；我想对模型进行量化（量化后），您能告诉我该怎么做吗？

Answer 1

要将检查点转换为 TFLite 模型，请参阅 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/r1/convert/python_api.md#convert-checkpoints-（来源：https://www.tensorflow.org/lite/convert）