search

patrickloeber / pytorch-chatbot

Simple chatbot implementation with PyTorch.
MIT License
420 stars 334 forks source link

Runtime error in train.py #3

Closed kenil84 closed 4 years ago

kenil84 commented 4 years ago

raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str))

RuntimeError: DataLoader worker (pid(s) 21184, 8440) exited unexpectedly

borhanreo commented 4 years ago

I think you run on your code in CPU machine. if yes, then you have to change two places in train.py 1: Change In line 84 num_workers=2 to num_workers=0 2: Add this after 98 line labels = labels.to(dtype=torch.long) (after labels = labels.to(device))