I want on/off the some filters in the layers on CNN.
what I want is below.
during test, i set the batch size to 1. then, the first batch enters network.
the network outputs the softmax values.
using this softmax values, I want to turn off the filters in network, which is strongly related to the low softmax values.
then next batch, off filter do not perform computation, and the number of softmax results is less than first batch.
can I do like above with pytorch?