Use average SGD