如何使用Node.js（tfjs-node）从Tensorflow.js中的检查点重新启动模型训练？

Question

我是训练模型，需要创建训练模型的检查点，然后从具有检查点的已保存级别重新开始模型训练。 tfjs-node有可能吗？如果是这样，请指导。

Answer 1

您可以使用model.save()方法并通过file：//网址保存模型，例如， await model.save('file:///tmp/my_model');。

稍后，您可以使用tf.loadModel()，也可以使用file：// URL，例如const model = await tf.loadModel('file:///tmp/my_model/model.json');来加载模型，所加载的模型保留先前训练的权重参数，并且可以在以后进行训练compile()通话。

请注意，当前TensorFlow.js的局限性在于它不会保存优化器的状态。因此，如果使用有状态优化器之一，例如“ adam”，“ rmsprop”，而不是默认的无状态优化器（即“ sgd”），则在保存和加载后，优化器的状态将被忘记。