search

chensong1995 / HybridPose

HybridPose: 6D Object Pose Estimation under Hybrid Representation (CVPR 2020)
MIT License
412 stars 64 forks source link

Training Time #42

Open ahmedreyad2019 opened 4 years ago

ahmedreyad2019 commented 4 years ago

How long does one epoch takes? it is very slow on my dataset, about 1 hour for every epoch. is that normal?

chensong1995 commented 4 years ago

Hi ahmedreyad2019,

Please consider setting num_workers>0 in torch.utils.data.DataLoader. This will speed up data loading.

Best, Chen