How to replace all ReLU activations in a pretrained network?

vision
8

Hi,

is the solution by @cassidylaidlaw also works for a loaded model that had nonlinearity in its forward call, and not in the init and using nn.Sequential?

Thanks!