gpu loss with torch no grad mix

dsssmplr