search

czg1225 / SlimSAM

SlimSAM: 0.1% Data Makes Segment Anything Slim
Apache License 2.0
248 stars 14 forks source link

Prune ratios for each step #12

Closed jameslahm closed 4 months ago

jameslahm commented 4 months ago

Thank you for your great work! Would you mind sharing the prune ratios for each step? I don't find them in the paper or code. Thanks!

czg1225 commented 4 months ago

Hi @jameslahm , For SlimSAM-50, pruning ratio is 0.5 for both step1 and step2, while for SlimSAM-77, pruning ratio is 0.77 for both step1 and step2. You can also change the pruning ratio if you want but I suggest that the pruning ratio in step 1 and the pruning ratio in step 2 should not be too different.

jameslahm commented 4 months ago

Thank you!