在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

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

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