implement varying weight for average SGD
Adjust average gradient descent