I understand, thanks.

is there a way to know all the buffers in a model?
in case of bn that you mentioned how I can get the mean and std values? I reach to this point, but not sure how to get  mean and std

import torchvision
import torch.nn as nn
import torch.nn.functional as F
resnet = torchvision.models.resnet50()
resnet.layer1[0].bn1.state_dict