Tensorflow NMT模型:恢复检查点并进行预测

时间:2019-07-13 16:27:56

标签: python python-3.x tensorflow machine-translation eager-execution

我已经训练了NMT Model TF-1.14.0,并保存了每个纪元的检查点。我想加载检查点并测试新的发音。我一直在寻找恢复检查点的方法,并发现'tf.train.Saver'可以保存会话和变量。但是代码示例未使用此方法。

我还检查了Tensorflow documentation - 1.14.0,并说使用了我在代码中完成的'tf.train.Checkpoint'。但是它没有显示如何加载检查点并预测新数据。有线索吗?

checkpoint_dir = './training_checkpoint'
checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")

checkpoint = tf.train.Checkpoint(optimizer=optimizer,
                           encoder=encoder,
                           decoder=decoder)

0 个答案:

没有答案