Learn PyTorch for deep learning in a day. Literally. #54

[danmoldashev]

I was completing the code step by step and got stuck on this moment

#plot_predictions(predictions=y_preds) # -> won't work... data not on CPU

#Put data on the CPU and plot it
plot_predictions(predictions=y_preds.cpu())

In my case .cpu() isn't helping to solve the converting from Tensor to numpy

[mrdbourke]

Hi @danmoldashev,

I've just tried this on my end and the notebook works running end to end.

Have you tried troubleshooting your code step by step?

See the original notebook here: https://github.com/mrdbourke/pytorch-deep-learning/blob/main/01_pytorch_workflow.ipynb