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

Question

這就是我保存 model 的方式

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)

在訓練時我在每個紀元之后都保存了這個

ckpt_manager.save()

后來恢復了model

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

這些是檢查點路徑中的文件

現在，這是我的問題； 我想量化模型（后量化），你能告訴我怎么做嗎？

Answer 1

要將檢查點轉換為 TFLite model，請參閱https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/r1/convert/python_api.md#convert-checkpoints- （來源： https：/ /www.tensorflow.org/lite/convert )