I am working on a modified resnet, and want to insert dropout after activation layers. I have tried this but due to the model not being sequential, it did not work: