Fix plot gradient log output and gradAtt computation, better use of device current velocity to initialize gradients