Trying to update the batchnorm.running_mean to a new value during training, where new_mean = old_mean * x + b
Seems like batchnorm parameters are not trainable and cannot be manipulated - is there a workaround to this?