I have a ResNet34 model and I want to find all the ReLU layer. I used named_modules() method to get the layers. for name, layer in ...
確定! 回上一頁