Fail to extract the last hidden layer from DistilBERT

16 views Asked by At

I want to extract the last hidden layer from the DistilBERT for sentiment analysis from huggingface IMDb dataset. I've tried several ways (at least I know...) to get the last hidden layer state; however, I always encounter the following error, please help me, thanks first!

enter image description here

enter image description here

0

There are 0 answers