Is there a way to train model on GPU without transformers? #7724
-
In the starter config from the main website, it seems there is no option to initialize the config for GPU (without transformers), there is only two options CPU, and GPU(transformers). If we want to train just regular model is there a sample config available showing how to do it? I tried disabling the transformers pipeline but it failed with an error during training. |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
You can pass 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. |
Beta Was this translation helpful? Give feedback.
You can pass
--gpu-id 0
(or your actual gpu id) to thespacy 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.