The code is failed to training

[YHDang]

Hello, thanks for your excellent research. I tried to reproduce your work but suffered from a question. The program stops while loading the DataLoader (line 98 in train_stage1.py). And it did not start to train. for ii, sample in enumerate(trainloader): I have tried to exchange the DATA_WORKERS to 0 but failed. Can you tell me some suggestions? Thank you very much

[sssdddwww2]

Would you show more details of the error information?

[YHDang]

Would you show more details of the error information?

Thanks for your reply. There is no error. The printed information is as follows. And then the program doesn't continue.

[lightas]

What's the version of pytorch you used? It is better to use pytorch 1.0.

[YHDang]

What's the version of pytorch you used? It is better to use pytorch 1.0.

Oh, I use pytorch 1.5. I'll try to change it to 1.0. Thanks very much.