THU-DA-6D-Pose-Group / GDR-Net

GDR-Net: Geometry-Guided Direct Regression Network for Monocular 6D Object Pose Estimation. (CVPR 2021)
https://github.com/THU-DA-6D-Pose-Group/GDR-Net
Apache License 2.0
266 stars 46 forks source link

Dataloader is quite slow #114

Open jiangranlv opened 8 months ago

jiangranlv commented 8 months ago

Hi,Dr Wang, nice work! I found that the IO speed is quite slow in the result that utility of GPU is relative low(~30). The GPU utilization rate will even become 0 every few seconds. The training time on ycbv is more than 26 hours. I whether is there any optimized strategy on this dataloader to speed up this process. Thank you!