Hi. I would like to extract all batch norm parameters from the pre-trained model?
May I know if you have any proper way to form a list of batch norm parameters?
This is due to the reason that I would like to retrain these parameters separately.
# for layer in resnet.modules():
#    if isinstance(layer,torch.nn.modules.batchnorm.BatchNorm2d): 
#        .....