RepViT icon indicating copy to clipboard operation
RepViT copied to clipboard

About distillation

Open muxin-wei opened this issue 1 year ago • 0 comments

Thanks for your great work! I've trained a SAM with TinyViT as image encoder, and am trying to replace it with Rep-ViT-M2.3, image_embedding of which is imposed to be the same. To gain a good performance, is it effective to just perform KD from TinyViT-SAM while copying and freezing the mask decoder following MobileSAM? Can you provide an example of how to perform KD with my trained SAM as the teacher model?

muxin-wei avatar Apr 02 '24 11:04 muxin-wei