fixing some issue with the regularization starting to overfit though
optimization server implemented