changing asgd to only average on the second epoch