thanks! I am trying to ensure that BatchNorm is used in training mode and is frozen because there are other layers that will be updated. Do I need to do this to the module after the net object is created??
net.bn.weight.requires_grad=False
net.bn.bias.requires_grad=False
net.bn.train()