search

microsoft / nlp-recipes

Natural Language Processing Best Practices & Examples
MIT License
6.35k stars 915 forks source link

[ASK] How to run on GPU #626

Open poko1 opened 1 year ago

poko1 commented 1 year ago

Description

I was running the codes for text classification (tc_mnli_transformers.ipynb) and it keeps on running on my cpu instead of gpu. How can I change that? It's taking way too long to train as a result. Please help.

Other Comments