PaddlePaddle/PaddleDetection

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

Open

#3.925 aberto em 10 de ago. de 2021

Ver no GitHub
 (4 comments) (0 reactions) (0 assignees)Python (2.731 forks)batch import
help wanted

Métricas do repositório

Stars
 (11.414 stars)
Métricas de merge de PR
 (Mesclagem média 2d 3h) (1 fundiu PR em 30d)

Description

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

Guia do colaborador