Actually, I want to implement ordered dropout as tackled in the fjord paper. The behavior of ordered dropout is dropping adjacent components of the model instead of random neurons. I am working on the Resnet18 model, and as a first step, I should override the forward method. Can someone give me some hints or ideas ?