Division by Zero - RUnning locally

I’m getting this error while training using the example in the course.
I’m running locally and using GPU. Installed the correct versions of libraries