r - 使用 MXnet 时如何保存模型

Question

我正在使用 MXnet 来训练 CNN(在 R 中)，我可以使用以下代码训练模型而不会出现任何错误:

model <- mx.model.FeedForward.create(symbol=network,
                                     X=train.iter,
                                     ctx=mx.gpu(0),
                                     num.round=20,
                                     array.batch.size=batch.size,
                                     learning.rate=0.1,
                                     momentum=0.1,  
                                     eval.metric=mx.metric.accuracy,
                                     wd=0.001,
                                     batch.end.callback=mx.callback.log.speedometer(batch.size, frequency = 100)
    )

但由于这个过程很耗时，我在晚上在服务器上运行它，我想保存模型，以便在完成训练后使用它。

我用了:

save(list = ls(), file="mymodel.RData")

和

mx.model.save("mymodel", 10)

但是他们都不能保存模型!例如，当我加载 "mymodel.RData" 时，我无法预测测试集的标签!

另一个例子是当我加载 "mymodel.RData" 时并尝试使用以下代码绘制它:

graph.viz(model$symbol$as.json())

我收到以下错误:

Error in model$symbol$as.json() : external pointer is not valid

任何人都可以给我一个保存然后加载这个模型以备将来使用的解决方案吗？

谢谢

Answer 1

您可以通过以下方式保存模型

model <- mx.model.FeedForward.create(symbol=network,
                                 X=train.iter,
                                 ctx=mx.gpu(0),
                                 num.round=20,
                                 array.batch.size=batch.size,
                                 learning.rate=0.1,
                                 momentum=0.1,  
                                 eval.metric=mx.metric.accuracy,
                                 wd=0.001,
                                 epoch.end.callback=mx.callback.save.checkpoint("model_prefix")
                                 batch.end.callback=mx.callback.log.speedometer(batch.size, frequency = 100)
)