RuntimeError - NPL with transformers book - 02_classification.ipynb

jgriffi · April 28, 2022, 11:54am

Hi,
I’m new to the Hugging Face library and getting started with NPL with transformers book. I’ve run the code in 02_classification.ipynb (section: Training the model) verbatim several times in Colab and got the same RuntimeError: Expected all tensors to be on the same device.

I can’t find any reference for this error and can’t see where I’m going wrong as the code was run my Colab is identical to notebooks/02_classification.ipynb at main · nlp-with-transformers/notebooks · GitHub

marshmellow77 · April 30, 2022, 4:13am

Hi Jay

This is being addressed in this Github issue: When running the Trainer cell, it found two devices (cuda:0 and CPU) · Issue #31 · nlp-with-transformers/notebooks · GitHub

There is a propsed workaround in the thread, hope that helps.

Cheers
Heiko

Topic		Replies	Views
RuntimeError: Expected all tensors to be on the same device Beginners	0	306	June 7, 2022
RuntimeError when running on Colab GPU 🤗Transformers	2	3277	November 28, 2021
HugginFace dataset error: RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor 🤗Datasets	3	8765	May 6, 2022
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select) Beginners	2	1492	July 30, 2023
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking arugment for argument index in method wrapper_index_select) Beginners	5	42081	July 4, 2023

RuntimeError - NPL with transformers book - 02_classification.ipynb

Related Topics