Range linear Quant-aware training: Require single GPU, and more

* Current implementation doesn't play nice with DataParallel, so allow
  only with single GPU for now
* Bug-fix: Update ranges only when training (not in eval)
* Refactor activation replacement function - explicit module instead of
  sequential