Incubator-mxnet: How to get the outputs of hidden layers when using Module for training

[sumethy]

I need to get the output of a hidden layer during inference. I use Module for training. After the training is done I can use

mod.forward(DataBatch)
output = mod.get_outputs() 

to get the output of the entire network. But I also want to inspect the output of a hidden layer. Is there any way to access that using Module's methods?

[bhokaal2k]

@sumethy Please checkout the tutorial for the same here
https://github.com/dmlc/mxnet-notebooks/blob/master/python/how_to/predict.ipynb

Read this tutorial fully to understand how to access the internal features and other cool features of MxNet, you will find your answer under the section "Extracting Features"

[sumethy]

@bhokaal2k Got it. Thanks