about the training time

Question

about the training time

Opened this issue 3 years ago · 4 comments

Thank you for sharing the code.
You mentioned in the article that you use a Titan X GPU to train the model. I want to know how much time the training process takes in the scannet dataset.
Looking forward to your reply.

Answer 1 · 2021-10-11T11:27:15.000Z

Thanks for your attention to our work. It takes 4 days to train HAIS for 500 epochs on a single Titan X GPU. Less epochs also achieve comparable performance. And switching to a better GPU can significantly speed up training.

Answer 2 · 2021-10-11T11:30:35.000Z

Thank for your reply.

Answer 3 · 2021-10-21T17:44:05.000Z

hi, to run the code, how much memory is needed?

Answer 4 · 2021-10-23T01:53:24.000Z

For batch size = 4, ~12G is OK.