Yeah, I don’t need to backprop the MI, just for feature selection.
I’ll probably do the feature selection with sklearn or something like that, I already have most of it.


Sounds like it should be quite easy.
Can you give a short code example of the changes in the forward part? let’s say a conv layer with 5 feature maps, a FC layer that follows, and selection of features according to [0,1,0,1,1].


Great.