lucidrains/vit-pytorch

CvT with 1 channel input data

tranlg99 opened this issue · 2 comments

tranlg99 commented

Hi, is it possible to input a 1 channel image into a CvT module? I see a channels argument for other modules but not for CvT

lucidrains commented

@tranlg99 sure, done

  • ❤️1
tranlg99 commented

Thank you!

  • ❤️1