The training time of baseline module?

JotaBravo / spacecraft-uda

Spacecraft Pose Estimation: Robust 2D and 3D-Structural Losses and Unsupervised Domain Adaptation by Inter-Model Consensus

https://ieeexplore.ieee.org/document/10225381

MIT License

19 stars 1 forks source link

The training time of baseline module? #11

Closed huangk77 closed 9 months ago

huangk77 commented 9 months ago

Hi! Thanks for your nice work.
I would like to know the training time of baseline module? I use the hourglass module to train the synthesis image of SPEED+, with batch size 4, and it take 2 hours each epoch. Is it normal for training?

JotaBravo commented 9 months ago

Hi, thanks for the interest.

Yes those are some normal times, the hourglass model is quite large and slow to train. Maybe you can see a slight improvement if you can use more cores for the data loader, but I'm afraid those are normal times.

Best,