setting grad None after training to avoid memory leak#1

kaloeffler wants to merge 1 commit into

kach:mainkach/gradient-descent-the-ultimate-optimizer:mainfrom

kaloeffler:mainkaloeffler/gradient-descent-the-ultimate-optimizer:mainCopy head branch name to clipboard