lucidrains / vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multi-target Regression Question

stethemJ opened this issue · comments

Hello,
I am using these vision transformers for single label regression and I am wondering if I would be able to use them for multiple label regression. I am not sure if I set the number of class to 2 if it will assume I am wanting classification over multi-target regression. Could anyone let me know?