I was searching for batchnorm layer and

model = models.densenet161()
for child in model.children():
  for layer in child.modules():
    if(isinstance(layer,torch.nn.modules.batchnorm.BatchNorm2d)):
      print(layer)

This code worked for me.