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

The process has been kill #89

Closed taylorlulu closed 8 months ago

taylorlulu commented 8 months ago

I encountered this problem when I'm running the command CUDA_VISIBLE_DEVICES=0 python train.py --camera realsense --log_dir logs/log_rs --batch_size 1 --dataset_root dataset/data/graspnet. Which says:

Loading grasping labels...: 100%|███████████████| 88/88 [00:28<00:00,  3.07it/s]
Loading data path and collision labels...:  91%|▉| 91/100 [00:17<00:02,  3.32it/
Killed

Which might be caused by the full use of the memory of swap. swap How can I solved this problem? I would appreciate a lot for any help!

chenxi-wang commented 8 months ago

That's caused by OOM. You could train the model on a subset or upgrade your machine.