小编Aja*_*jai的帖子

我一直在尝试从 onnx 格式导入模型以使用 pytorch。我发现很难找到相同的例子。由于 Internet 上的大多数资源都在谈论将 pytorch 模型导出到 onnx。

发现torch.onnx()只能导出模型，还没有实现导入方法。直接安装 onnx 库，帮助我做 onnx.load("model_name.onnx")。我如何在 pytorch 中使用这个模型？我无法使用 model.to(device="GPU") 将模型移动到 GPU