I want to print how parameters are changed after batchnorm.
In my model __init__
self.fc0 = nn.Linear(4096, 300)
self.bn0 = nn.BatchNorm1d(300)
def forward():
x = self.fc0(x)
x = self.bn0(x)
then how can I print parameters after each step?