[参考译文] PROCESSOR-SDK-AM62A：如何在 mmyolo 存储库下生成 pt 文件

尊敬的 Ryuuuichi：

 https://github.com/TexasInstruments/edgeai-tensorlab/issues/7中步骤的最终目的 是导出.onnx 文件以移植到 TIDL ( https://e2e.ti.com/support/processors-group/processors/f/processors-forum/1457711/faq-am62a7-q1-deploy-mmyolo-on-am62a-yolov8-nano-as-example 中包含更多分步详细信息)。

目前、.pth 文件用于上述目的、因此不打算改用.pt。  

我可以问为什么在您的用例中首选.pt？

此致、

Qutaiba

您好：

感谢您的答复。

 https://github.com/TexasInstruments/edgeai-tensorlab/issues/7中步骤的最终目的 是导出.onnx 文件以移植到 TIDL ( https://e2e.ti.com/support/processors-group/processors/f/processors-forum/1457711/faq-am62a7-q1-deploy-mmyolo-on-am62a-yolov8-nano-as-example 中包含更多分步详细信息)。
=>是的,我理解上面。

>我可以问为什么在您的使用案例中首选.pt？
我们希望通过使用具有相同模型的其他平台来制定基准。
如您所述、TI 平台需要 pth、但我们将使用的其他平台需要 pt 文件。 我问这个。

此致、

尊敬的 Ryuuuichi：

感谢您提供详细信息。 我的理解是、.pth 和.pt 可以在现代 Pytorch 版本中互换使用。

目前,我只能提供一个实验建议:

• 尝试手动更改所生成文件的扩展名。
• 如果上述操作不起作用、我建议查找 https://github.com/open-mmlab/mmyolo 用于保存模型的代码并添加另一行、如 tolk.save (model、path/name.pt)。  

此致、

Qutaiba

您好：

我正在尝试执行您的想法2.

• 如果上述操作不起作用、我建议查找 https://github.com/open-mmlab/mmyolo 用于保存模型的代码并添加另一行、如 tolk.save (model、path/name.pt)。  

首先、我正在检查执行"torch.save"的代码。
我知道以下文件执行保存文件(proce_checkpoint (in_file、out_file))。
mmyolo/.dev_scripts/main·open-mmlab/mmyolo·GitHub 上的 gather_models.py

但是、"out_file"似乎来自"training result"。
我不确定哪个文件执行训练本身。 似乎在培训课程中给出了"第"个字。
所以,我认为我需要在培训课程中改变性格。

但我找不到执行此操作的文件。 您是否知道哪个文件执行培训？

此致、