Skip to content
Discussion options

You must be logged in to vote

You can pass --gpu-id 0 (or your actual gpu id) to the spacy train command to train on GPU.

I think the reason that this isn't included in the start configs is that if you have a GPU, the boost in accuracy from Transformers is generally worth using.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by svlandeg
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
gpu Using spaCy on GPU feat / transformer Feature: Transformer
2 participants