Solved by JuanFMontesinos in post #6 
              
              
                They key is the name that you assign to the variable in the nn.Module, therefore 
class test(torch.nn.Module):
    def __init__(self):
        super(test,self).__init__()
        self.conv1 = torch.nn.Conv2d(10,15,10)
        self.customconv = torch.nn.Conv2d(100,1000,10)

test()
Out[7]: 
test(
  (c…