pretraining 96k CT scan

BAAI-DCAI / SegVol

The official code for "SegVol: Universal and Interactive Volumetric Medical Image Segmentation".

MIT License

245 stars 22 forks source link

Closed winwinJJiang closed 2 weeks ago

winwinJJiang commented 1 month ago

Hi,

How did you end up with pretrain your 96k CT scan for 2000 epochs with only 8*40GB GPU? How long it takes?

Our experience is that this is practically impossible. We have even more GPUs (32*80GB) but pretraining for 500 epoch has token 7 days.

Any idea or clarification?!?!?!?!?!!?

Yuxin-Du-Lab commented 2 weeks ago

Thanks for your inquiry. Your question has been answered in issue #25