我已经训练了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)