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