Hi @python-engineer , on line 131, shouldn't we assign z again on CPU? z = z.to("cpu") before then calling z.numpy() I got following error without doing so: <pre><code>TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.</pre></code> Thanks for all the great material! Cheers, Gautam