search

PKU-EPIC / UniDexGrasp

Official code for "UniDexGrasp: Universal Robotic Dexterous Grasping via Learning Diverse Proposal Generation and Goal-Conditioned Policy" (CVPR 2023)
136 stars 13 forks source link

I have a question about training time #16

Closed le-wei closed 4 months ago

le-wei commented 11 months ago

Hello, I have a question about training time. I'm using an RTX 3080 with 10GB of VRAM for training, and it takes 3 to 6 hours to complete one epoch. Is this correct? Thank you for your response.

error3

error4

mzhmxzh commented 11 months ago

May I ask which module you are training? Is it the GraspIPDF or GraspGLOW or ContactNet?

le-wei commented 11 months ago

"Currently, I have tested GraspIPDF and GraspGlow using the commands 'python ./network/train.py --config-name ipdf_config --exp-dir ./ipdf_train' and 'python ./network/train.py --config-name glow_config --exp-dir ./glow_train,' respectively. Both take approximately the same amount of time. Is there something that needs to be modified?"