jiecaoyu / XNOR-Net-PyTorch

PyTorch Implementation of XNOR-Net
472 stars 119 forks source link

Batch Normalisation #96

Closed kannan500 closed 4 years ago

kannan500 commented 4 years ago

https://github.com/jiecaoyu/XNOR-Net-PyTorch/blob/ebccb503b2f9ebc96187a75b6daa63d95b61d110/ImageNet/datasets/transforms.py#L58 In this line,you are describing Normalisation ,but I have a doubt... @jiecaoyu How you find mean and variance during inference?

kannan500 commented 4 years ago

In inference,Are you using the mean and variance from training ?

jiecaoyu commented 4 years ago

The mean and variance are computed using the script described here. They will not change through the training.