Error while using it in kaggle

#4
by rohitchaudhari25 - opened

I get below error after i tried usign this model on kaggle

Some weights of ViTModel were not initialized from the model checkpoint at /kaggle/input/facebookdino-vitb8/transformers/default/1 and are newly initialized: ['pooler.dense.bias', 'pooler.dense.weight']
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.

I want to use this model to extract pooler output as feature vector of image for downstream classification task.

image.png

Sign up or log in to comment