Hello, I am wondering that the moving_mean and moving_variance in the 'bn' layers seem no to be updated by this code.