cvlab-stonybrook / SAMPath

Repository for "SAM-Path: A Segment Anything Model for Semantic Segmentation in Digital Pathology" (MedAGI2023, MICCAI2023 workshop)
30 stars 6 forks source link

Some problems about SAMPath #8

Closed NanCheng2001 closed 8 months ago

NanCheng2001 commented 8 months ago

Thank you for your outstanding work!

I would like to ask, what are the equipment requirements for your model training, and at least how much GPUmemory is required? Because your article does not specify which parameters are frozen and which parameters need to be retuned.

jingweizhang-xyz commented 8 months ago

I did not remember the exact numbers. I think it should be <10GB for given batch size. The network is very light-weighted. Because the feature backbone is frozen and the rests are fine tuned but they do not have many parameters.

NanCheng2001 commented 8 months ago

Thank you for your patience.