在MxNet中如何保存训练器的内部变量?

我编写了一个包含分布式随机梯度下降(DRGD)算法的程序。该算法中有一些内部变量,用于计算步长。训练算法可能比DRGD复杂得多,因此会有更多的内部变量。如果我们保存这些变量,就可以暂停训练并测试模型;然后,我们可以再次恢复训练。


回答:

在使用MXNet Gluon进行训练时,可以通过调用Trainer类的.save_states().load_states()函数来保存训练器的状态并恢复训练。

这是一个示例:

trainer = gluon.Trainer(net.collect_params(), 'adam')
trainer.save_states('training.states')
trainer.load_states('training.states')

Related Posts

Keras Dense层输入未被展平

这是我的测试代码: from keras import…

无法将分类变量输入随机森林

我有10个分类变量和3个数值变量。我在分割后直接将它们…

如何在Keras中对每个输出应用Sigmoid函数?

这是我代码的一部分。 model = Sequenti…

如何选择类概率的最佳阈值?

我的神经网络输出是一个用于多标签分类的预测类概率表: …

在Keras中使用深度学习得到不同的结果

我按照一个教程使用Keras中的深度神经网络进行文本分…

‘MatMul’操作的输入’b’类型为float32,与参数’a’的类型float64不匹配

我写了一个简单的TensorFlow代码,但不断遇到T…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注