MXnet affiner sauver modèle

Question

J'utilise l'exemple-tune fin de mxnet pour affiner mes propres données avec ce code:

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

En visualisant commune/fit.py, je me suis aucune idée de la façon d'enregistrer le modèle de température lorsque je peaufiner. Par exemple, je veux enregistrer des fichiers .params tous les 5000 iters, comment puis-je le faire? THX!

Answer 1

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

Essayez d'utiliser l'API mx.callback.

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"