Gradient Save and Restart Fix