ChineseBert icon indicating copy to clipboard operation
ChineseBert copied to clipboard

关于CPU

Open SharelockSmile opened this issue 4 years ago • 5 comments

我想请问一下,在运行tasks中的LCQMC示例的时候,如果没有GPU该如何调整

python LCQMC_trainer.py --bert_path ../ChineseBERT-base/ --data_dir E:/PycharmProjects/ChineseBert/lcqmc/ --save_path ../../cus_output/ --max_epoch=7 --lr=2e-5 --batch_size=16 --gpus=0最后一个参数可以不写吗?需要如何调整参数和修改代码

SharelockSmile avatar Jul 21 '21 08:07 SharelockSmile

LCQMC_trainer.py使用了pytorch_lightning框架,参数的输入参照该框架要求。52行和149行可能需要做相应改动。 建议使用GPU, CPU训练会非常慢。。

zijunsun avatar Jul 21 '21 09:07 zijunsun

LCQMC_trainer.py使用了pytorch_lightning框架,参数的输入参照该框架要求。52行和149行可能需要做相应改动。 建议使用GPU, CPU训练会非常慢。。

本机上目前还没有GPU,想先测试一下,就可以不写--gpus这个参数吗

SharelockSmile avatar Jul 21 '21 09:07 SharelockSmile

可以的,但是代码也要相应做一些改动。

zijunsun avatar Jul 22 '21 09:07 zijunsun

可以的,但是代码也要相应做一些改动。

我改完之后又报了新的错,这和我的配置有关系吗,是用CPU运行的, 报错

SharelockSmile avatar Jul 22 '21 09:07 SharelockSmile

cpu下你试下num_workers设置为0.

zijunsun avatar Jul 22 '21 13:07 zijunsun