search

ethnhe / PVN3D

Code for "PVN3D: A Deep Point-wise 3D Keypoints Hough Voting Network for 6DoF Pose Estimation", CVPR 2020
MIT License
482 stars 105 forks source link

算力要求 #99

Open aoligei178 opened 2 years ago

aoligei178 commented 2 years ago

请问训练pvn3d在linemod数据集上要多少显卡？

HP-CAO commented 2 years ago

I tested training pvn3d with batchsize of 1 on a single GTX 1080 card. It was almost out of the memory. I think if you want to train with batchsize of 16, you need to get a nice graphical card like A100, v100 or use multi GPUs (4xGTX 1080). And maybe you can just train a tiny pvn3d with fewer parameters such that you can run with a larger batchsize. To get the decent result, I personally think 48G+ memory is necessary.

aoligei178 commented 2 years ago

Thank you very much! I have another question. Could I know more specifically about your training process? How much time you used before you got final parameters?

gaoyangithub commented 1 year ago

请问训练pvn3d在linemod数据集上要多少显卡？

你好，python-pcl一直安装不上，显示没有，但是已经离线安装，怎么解决呢？