search

hemingkx / CLUENER2020

A PyTorch implementation of a BiLSTM\BERT\Roberta(+CRF) model for Named Entity Recognition.
470 stars 107 forks source link

请问如何使用gpu来训练呢？ #17

Open Siomarry opened 1 year ago

Siomarry commented 1 year ago

网上说的使用gpu训练有几个地方都需要设置，数据集、损失函数、模型. 代码里面模型的设置. 请问代码可以在gpu环境下运行吗