Retrain SWIN transformer to predict a new class

#1
by abdulsalama - opened

Hi all, I'm trying to retrain a swin transformer for instance segmentation to predict new class. I'm looking at this pretrained model "facebook/maskformer-swin-small-coco". I'm relatively new to Hugging Face. Any hint or pointer if this is possible and how can I do that? Would really appreciate any help! Thanks!

Sign up or log in to comment