grad_comparison is beefier now, not quite done. bug fix to training, counting issue
Merge branch 'master' of https://github.com/deba2000/CS250B