Sorry, let me clarify.
I have network components defined in blocks like
self.block1 = nn.Sequential(
    nn.Conv1d(3,64,1),
    nn.BatchNorm1d(64),
    nn.ReLU())

How can I modify the momentum of those batch norm layers in the network?