james34602 / SpleeterRT

Real time monaural source separation base on fully convolutional neural network operates on Time-frequency domain.
GNU General Public License v3.0
157 stars 16 forks source link

about batch normalization #2

Closed yansd-c closed 4 years ago

yansd-c commented 4 years ago

Great works! I read your code, but I am confused about the batch norm operation. I think batch norm in model prediction is (x-moving_mean)/movingvar * gamma + beta., while in the code is batchNorm[ + s] * val + batchNorm[ _ ]. I am not sure if it is correct.

james34602 commented 4 years ago

Hello The batch normalization equation was heavily modified for the model(Especially follow the way official Tensorflow Spleeter), which simplify computation.