tests/test_sim_bn_fold.py · 4c7d4890eb27c56bb6061ed4c859f48b679b0e49 · llvm / distiller

5 years ago

Simulated BN folding during training (module impl only) (#274) · 4c7d4890

Lev Zlotnik authored 5 years ago

Implementation of simulated BatchNorm folding as per
https://arxiv.org/pdf/1806.08342.pdf

* NOTE: This is just the module implementation for now, not yet integrated
  with QuantAwareTrainRangeLinearQuantizer

4c7d4890

History

Simulated BN folding during training (module impl only) (#274)

Lev Zlotnik authored 5 years ago

Implementation of simulated BatchNorm folding as per
https://arxiv.org/pdf/1806.08342.pdf

* NOTE: This is just the module implementation for now, not yet integrated
  with QuantAwareTrainRangeLinearQuantizer