我正在训练模型，需要创建训练模型的检查点，并从带有检查点的保存级别重新启动模型训练。 tfjs-node 可以做到这一点吗？如果是的话请指导..
最佳答案
您可以使用 model.save() 方法保存模型，并使用 file://URL，例如，
等待 model.save('file:///tmp/my_model');。
稍后，您可以使用 tf.loadModel() 加载模型，也可以使用 file://URL，例如 const model = wait tf.loadModel('file:///tmp/my_model/model.json'); 加载的模型保留了之前训练的权重参数，并且可以在 compile() 调用之后进一步训练。
请注意，目前 TensorFlow.js 存在一个限制，即它不保存优化器的状态。因此，如果您使用有状态优化器之一，例如“adam”、“rmsprop”，而不是默认的无状态优化器(即“sgd”)，则在保存和加载后将忘记优化器的状态。
关于javascript - 如何使用 Node.js (tfjs-node) 从 Tensorflow.js 中的检查点重新启动模型训练？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/54339478/