pytorch - 在 C++ 中使用 *.pth 模型

Question

我想使用我用 pytorch 训练的 yolo3 模型在 C++ 中运行推理。我无法使用 pytorch 提供的跟踪和脚本进行转换。我在转换过程中遇到此错误

First diverging operator:
    Node diff:
        - %2 : __torch__.torch.nn.modules.container.ModuleList = prim::GetAttr[name="module_list"](%self.1)
        + %2 : __torch__.torch.nn.modules.container.___torch_mangle_139.ModuleList = prim::GetAttr[name="module_list"](%self.1)
        ?                                           ++++++++++++++++++++
ERROR: Tensor-valued Constant nodes differed in value across invocations. This often indicates that the tracer has encountered untraceable code.
    Node:
        %358 : Tensor = prim::Constant[value=<Tensor>](), scope: __module.module_list.16.yolo_16