search

PaddlePaddle / models

Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
Apache License 2.0
6.89k stars 2.91k forks source link

Paddle1.6 模型库 paddleNlp/sentiment_classification 这个怎么单机多卡训练啊 #4637

Open huoyuming opened 4 years ago

huoyuming commented 4 years ago

Paddle1.6 模型库 paddleNlp/sentiment_classification
run_ernie.sh 设置了环境变量 CUDA_VISIBLE_DEVICES=0,1,2,3,4,5 但是实际跑起来只有0在跑，其他的都没有跑 这个怎么单机多卡训练啊

guoshengCS commented 4 years ago

可以试下参照这里 https://github.com/PaddlePaddle/models/blob/develop/PaddleNLP/machine_translation/transformer/train.py#L164 ，将这里 https://github.com/PaddlePaddle/models/blob/develop/PaddleNLP/sentiment_classification/run_classifier.py#L258 train_exe.run使用的program改为CompiledProgram

huoyuming commented 4 years ago

可以试下参照这里 https://github.com/PaddlePaddle/models/blob/develop/PaddleNLP/machine_translation/transformer/train.py#L164 ，将这里 https://github.com/PaddlePaddle/models/blob/develop/PaddleNLP/sentiment_classification/run_classifier.py#L258 train_exe.run使用的program改为CompiledProgram

好的 谢谢了 我试一下~~