The training time of baseline module?

Question

The training time of baseline module?

huangk77 opened this issue a year ago · 1 comments

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?

Answer 1 · 2023-10-09T12:38:16.000Z

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,