Tweaks to restrict omega for parallel SGD