modify adagrad to leverage good initializations