PaddlePaddle/PaddleDetection

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

Open

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

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

仓库指标

Star
 (11,414 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

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

贡献者指南

研究方向
研究当模型作为参数传递给回调函数时，PaddlePaddle 如何处理 GPU 上下文。检查模型的 GPU 上下文是否丢失，或者是否需要在回调函数内显式地将模型移动到 GPU。查阅 PaddlePaddle 文档以及关于回调和 GPU 使用的现有问题。
技术栈
python
领域
backend
议题类型
缺陷
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 小时
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
需要先调研
前置要求
PythonPaddlePaddle
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
50