MXnet微调保存模型

Question

我正在使用mxnet的微调示例来使用此代码微调我自己的数据：

https://github.com/dmlc/mxnet/blob/master/example/image-classification/fine-tune.py

通过查看common / fit.py，我在微调时不知道如何保存临时模型。

例如，我想每5000个文件保存.params文件，我该怎么办？ THX！

Answer 1

http://mxnet.io/api/python/callback.html

尝试使用mx.callback API。

module.fit(iterator, num_epoch=n_epoch,
... epoch_end_callback  = mx.callback.do_checkpoint("mymodel", 1))
Start training with [cpu(0)]
Epoch[0] Resetting Data Iterator
Epoch[0] Time cost=0.100
Saved checkpoint to "mymodel-0001.params"
Epoch[1] Resetting Data Iterator
Epoch[1] Time cost=0.060
Saved checkpoint to "mymodel-0002.params"