Export mxnet network in training mode

[AminAnsarian]

Hello all,

I have a network which behaves differently (different number of outputs) during training and prediction (when autograd.is_training() is true and false). By default, mxnet.export exports the model in prediction mode and assumes autograd.is_training() to be false. Does anybody know if it is possible to export the network in training mode as well?

Thank you.