PaddlePaddle/PaddleDetection

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

オープン

#3,925 opened on 2021/08/10

 (4 件のコメント) (0 件のリアクション) (0 人の担当者)Python (2,731 件のフォーク)batch import
help wanted

Repository metrics

Stars
 (11,414 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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

コントリビューターガイド