NVlabs / DG-Net

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:
https://www.zdzheng.xyz/publication/Joint-di2019
Other
1.27k stars 230 forks source link

when I run train.py, print there error message. Can you help to solve the issue? Thank you ! #72

Open yuanqs opened 3 years ago

yuanqs commented 3 years ago

Every epoch need 404 iterations Note that dataloader may hang with too much nworkers. Traceback (most recent call last): File "", line 1, in File "C:\Users\hcytech\anaconda3\envs\Python38_yolos\lib\multiprocessing\spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "C:\Users\hcytech\anaconda3\envs\Python38_yolos\lib\multiprocessing\spawn.py", line 125, in _main prepare(preparation_data) File "C:\Users\hcytech\anaconda3\envs\Python38_yolos\lib\multiprocessing\spawn.py", line 236, in prepare _fixup_main_from_path(data['init_main_from_path']) File "C:\Users\hcytech\anaconda3\envs\Python38_yolos\lib\multiprocessing\spawn.py", line 287, in _fixup_main_from_path main_content = runpy.run_path(main_path, File "C:\Users\hcytech\anaconda3\envs\Python38_yolos\lib\runpy.py", line 265, in run_path

layumi commented 2 years ago

For windows user, you need to set the workers in data loader as 0. It is due to the pytorch multi-processing tool, which is not optimized for windows.