Cannot extract penultimate layer output of a vision transformer with pytorch
I have the following model that I tuned with my own dataset trained with DataParallel
I have the following model that I tuned with my own dataset trained with DataParallel