【问题标题】:How to restart a model training from a check point in Tensorflow.js with Node.js (tfjs-node)?如何使用 Node.js(tfjs-node)从 Tensorflow.js 中的检查点重新开始模型训练?
【发布时间】:2019-01-24 04:40:58
【问题描述】:

我正在训练模型,需要创建训练模型的检查点,并从保存的带有检查点的级别重新开始模型训练。这可能与 tfjs-node 吗?如果是请指导..

【问题讨论】:

    标签: javascript node.js tensorflow machine-learning tensorflow.js


    【解决方案1】:

    您可以使用 model.save() 方法保存模型,并带有 file:// URL,例如, 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”),优化器的状态将在保存和加载后被遗忘。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2020-03-16
      • 2021-09-11
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2023-01-03
      • 2020-12-13
      • 2020-04-06
      相关资源
      最近更新 更多