deep-learning-with-pytorch / dlwpt-code

Code for the book Deep Learning with PyTorch by Eli Stevens, Luca Antiga, and Thomas Viehmann.

Home Page:https://www.manning.com/books/deep-learning-with-pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Dataloader worker quit unexpectedly

cmoore976 opened this issue · comments

I found a problem when running the default training code in P2ch11. If you do not have enough system resources, the program will stop with the above error. You need to lower your number of workers and batch size in order to get the file to run. This will variate according to how much system ram and vram you have.

Oh, yes. Sorry, we should have been more diligent in explaining this. I will likely try to take a stab at updating the code to the latest PyTorch soonish.

thanks. have a great weekend!!