graspnet / graspnet-baseline

Baseline model for "GraspNet-1Billion: A Large-Scale Benchmark for General Object Grasping" (CVPR 2020)
https://graspnet.net/
Other
475 stars 142 forks source link

I meet a question that the train process can only run when the num_workers of Dataloader is 0. The memory is 32G. Ubuntu20.04. Pytorch1.10. I have found that the memory usage is gradually increasing during loading grading labels, as well as loading data path and consolidation labels, due to insufficient memory, resulting in num_ Must workers be 0 to run? When this parameter is 0, the training is slow. I hope someone can help. #67

Closed wwk-png closed 1 year ago

wwk-png commented 1 year ago

I use a computer with 32GB of memory and 32GB of swap space. Do I need larger memory to support training?

Fang-Haoshu commented 1 year ago

Hi, to speed up, a larger memory is needed. Perhaps you need a memory larger than 128GB