Just added my code (see above) as I can’t get the jupyter notbook to work on my github.
(Always stops rendering with “something went rong”)
My output is:

Time passed on CPU (Batch Gradient Descent) 0.20313024520874023 s
Time passed on GPU (Batch Gradient Descent) 15.053136348724365 s


Time passed on CPU (Stochastic Gradient Descent) 0.015608549118041992 s
Time passed on GPU (Stochastic Gradient Descent) 1.3855891227722168 s