HAIS icon indicating copy to clipboard operation
HAIS copied to clipboard

about the training time

Open WangZhouTao opened this issue 4 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.

WangZhouTao avatar Oct 11 '21 08:10 WangZhouTao

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.

outsidercsy avatar Oct 11 '21 11:10 outsidercsy

Thank for your reply.

WangZhouTao avatar Oct 11 '21 11:10 WangZhouTao

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

souri1234 avatar Oct 21 '21 17:10 souri1234

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

outsidercsy avatar Oct 23 '21 01:10 outsidercsy