Thanks for your reply!
I tried to update the momentum with the following code:
        for subnetwork in self.children():
            for layer in subnetwork.children():
                if type(layer) == torch.nn.BatchNorm1d:
                    layer.momentum = a * b**(self.epoch/c)

which seems to work.