Solved by reachtarunhere in post #2 
              
              
                Do you plan to treat Conv1d, Conv2d and so far as different? If you were only looking for Conv2d layers you can do something like: 
for layer in net.children():
    if isinstance(layer, nn.Conv2d):
        do something with the layer

isinstance is a Python built-in https://docs.python.org/3/library…