RepViT
RepViT copied to clipboard
About distillation
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?