search

lukemelas / PyTorch-Pretrained-ViT

Vision Transformer (ViT) in PyTorch
770 stars 124 forks source link

Multilabel Image classification #21

Open abhigoku10 opened 3 years ago

abhigoku10 commented 3 years ago

@lukemelas thanks for sharing code base , i had few queries 1.can we train crossvit for multilabel classification problem , if so what is the procedure

  1. i have a custom dataset of 10.5k with 25class labels with instance as label vectors of 0 and 1
  2. can remove the pre-trained classifer head and add our customr classifier ?

Thanks in advance