Tensorflow NMT模型:恢复检查点和预测

2024-10-16 20:43:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经训练了NMT ModelTF-1.14.0并为每个历元保存了检查点。我想加载检查点并测试新的语句。我正在寻找恢复检查点的方法,发现tf.列车保护器“保存会话和变量。但是代码示例不使用此方法。你知道吗

我还检查了Tensorflow documentation - 1.14.0,它说使用'列车检查站“我在代码中已经做了。但它没有显示如何加载检查点和预测新数据。有线索吗?你知道吗

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

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

Tags: 方法代码encodertfdir语句检查点optimizer