pytorch - 无法将 Pytorch 模型导出到 ONNX

Question

我正在尝试将经过训练的模型以 .pt 格式导出到 ONNX，但是在执行脚本后，程序输出一堆日志记录内容，如下图所示，并且转换后的 onnx 格式似乎没有出现在我的本地磁盘中我没有知道我在这里做错了什么吗？

score 0 · Accepted Answer

看起来出口很顺利。在您运行脚本的目录中应该有一个文件“ScaledYolo4.onnx”。

要确定它是什么目录，您可以在程序末尾添加：

import os
print(os.getcwd())

您也可以尝试指定完整路径，而不仅仅是“ScaledYolo4.onnx”。

编辑. 在程序末尾添加这个并运行，你看到了什么？

import os
print(os.getcwd())
print(os.listdir())

1 回答 1