我正在使用 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)
)
但由于这个过程很耗时,所以我晚上在服务器上运行它,我想保存模型以便在训练完成后使用它。
I used:
save(list = ls(), file="mymodel.RData")
and
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
有人可以给我一个保存然后加载该模型以供将来使用的解决方案吗?
Thanks