Huggingface transformer sequence classification

🤗Transformers
4

Hi,

You shouldn’t actually have any device placements, the Trainer handles this for you. The Trainer has an attribute called place_model_on_device which is set to True by default.

The Trainer will also handle device placement of your data.