Quick fixes: prints out epoch, batch for nan error; correctly handles hessian calc when on gpu
Quick fixes: prints out epoch, batch for nan error; correctly handles hessian calc when on gpu

billbrod/spatial-frequency-preferences,billbrod/spatial-frequency-preferences,billbrod/spatial-frequency-preferences