HAIS
HAIS copied to clipboard
about the training time
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.
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.
Thank for your reply.
hi, to run the code, how much memory is needed?
For batch size = 4, ~12G is OK.