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

Question

我正在训练模型，需要创建训练模型的检查点，并使用检查点从保存的级别重新启动模型训练。tfjs-node可以做到这一点吗？如果有，请指点一下..

Answer 1

您可以使用model.save()方法以文件:// URL保存您的模型，例如await model.save('file:///tmp/my_model');。

稍后，您可以使用tf.loadModel()加载模型，也可以使用文件:// URL，例如，const model = await tf.loadModel('file:///tmp/my_model/model.json');加载的模型保留了以前训练的权重参数，并可以在compile()调用后进行进一步训练。

请注意，目前TensorFlow.js有一个限制，因为它不保存优化器的状态。因此，如果您使用其中一个有状态优化器，例如'adam'，'rmsprop'，而不是默认的无状态优化器(即'sgd')，优化器的状态将在保存和加载后被遗忘。