Hey,
I’m new to Pytorch. I’m trying to implement a compressed network where I switch off a neuron  after every few epochs throughout training based on some threshold criteria(something similar to dropout). I’m not sure how to implement this in forward if that’s where masking of a neuron should be done. Thanks in advance.