PaddlePaddle/PaddleDetection

模型对象作为参数传入到回调函数里无法调用GPU?

Open

#3925 aperta il 10 ago 2021

Vedi su GitHub
 (4 commenti) (0 reazioni) (0 assegnatari)Python (2731 fork)batch import
help wanted

Metriche repository

Star
 (11.414 star)
Metriche merge PR
 (Merge medio 2g 3h) (1 PR mergiata in 30 g)

Descrizione

业务需求中存在回调函数,数据源来源回调函数,因此需要把模型初始化放在主函数,推理部分通过传参进入回调函数进行推理,这样做代码可以跑,但是无法调用GPU,把推理函数放到主函数内推理就可以调用GPU,因此不知道如何解决在回调函数内进行推理无法调用GPU的问题如何解决?

Guida contributor