Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0 – Text Classification
I use ‘bert-base-uncased’ model. I fine-tuned my dataset using BERT model and saved it in my Drive. Then I load the trained model and test it.