try switching to ADAGrad regularized trainer