baidu/Senta

如何指定单卡预测。

hyv533 opened this issue · 1 comments

Senta().init_model(model_class="ernie_1.0_skep_large_ch", task="sentiment_classify", use_cuda=use_cuda)

默认是在device: 0上,如何指定gpu。

同问