PaddlePaddle/PaddleDetection

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

Open

#3,925 创建于 2021年8月10日

在 GitHub 查看
 (4 评论) (0 反应) (0 负责人)Python (11,414 star) (2,731 fork)batch import
help wanted

描述

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

贡献者指南

技术栈
python
领域
machine learning
议题类型
bug
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 hours
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
stale
清晰度议题是否清楚说明期望改动、验收标准和下一步。
needs investigation
前置要求
basic PaddlePaddle knowledgeGPU acceleration basics
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
25
研究方向
Investigate how PaddlePaddle handles GPU context when a model is passed as an argument to a callback function. Check if the model's GPU context is lost or if there is a need to explicitly move the model to GPU within the callback. Look at PaddlePaddle documentation and existing issues related to callbacks and GPU usage.