NVlabs/NViT

How to do downstream tasks transfer learning?

Opened this issue · 0 comments

I want to do downstream tasks transfer learning on CIFAR-10 and CIFAR-100, but the error information is as follows: RuntimeError: The size of tensor a (100) must match the size of tensor b (1000) at non-singleton dimension 1. Could you help me with this error? Thank you very much!