CrossViT-pytorch icon indicating copy to clipboard operation
CrossViT-pytorch copied to clipboard

Multilabel Image classification

Open abhigoku10 opened this issue 4 years ago • 0 comments

@rishikksh20 thanks for sharing code base , i had few queries 1.can we train crossvit for multilabel classification problem , if so what is the procedure 2. i have a custom dataset of 10.5k with 25class labels with instance as label vectors of 0 and 1 3. can remove the pre-trained classifer head and add our customr classifier ?

Thanks in advance

abhigoku10 avatar Aug 30 '21 17:08 abhigoku10