search

IGLICT / SketchDream

Apache License 2.0
52 stars 4 forks source link

problems about num_workers setting #8

Closed Lanjiong-Li closed 2 weeks ago

Lanjiong-Li commented 2 weeks ago

when I run ./scripts/mushroom.sh , it prints a warning: The 'train_dataloader' does not have many workers which may be a bottleneck. Consider increasing the value of the num_workers argumenttonum_workers=223in theDataLoader` to improve performance. I wonder where can I modify numworkers? I guess I can find it here SketchDream/threestudio/data, but there are lots of scripts in this folder, should I modify all of them? I also see some comments like # Multi-process, more quickly, with small bug now_ in SketchDream/threestudio/data/random_multiview_control_depth.py. Is it relates to this problem?

ty625911724 commented 2 weeks ago

We currently do not suggest to use multi-process with num_worker>0 for dataloader. This problem is inherited from three-studio. If directly setting num_worker>0, when changing the image render size or other settings, it will not work. I have tried to utilize some synchronous tips, but it occasionally caused zombie process. Those comments are related to it and you can try it by yourself.