search

SamsungLabs / ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation
MIT License
626 stars 125 forks source link

Training time #3

Closed UndecidedBoy closed 3 years ago

UndecidedBoy commented 3 years ago

Hello,

Great work and GitHub repository! I'm curious about the duration needed to complete training, was it a matter of days or weeks (for the Nvidia Tesla P40 GPUs you used)?

Best!

ksofiyuk commented 3 years ago

It takes about 2-3 days to train the HRNet18 IT-M model on COCO+LVIS using two Tesla P40.

UndecidedBoy commented 3 years ago

Thanks for the response.