看这个现象,应该不是你操作问题,像是当前 RUHMI 这条工具链暂时不直接吃 `.pth` 了,所以选项才会灰掉。`.pth` 更偏 PyTorch 训练阶段的模型权重文件,真正部署到 MCU/NPU 端,一般还是要转成工具链能解析的推理格式。
我会优先走 .pth -> ONNX -> TFLite 这条路线,中间用 ONNX 先把网络结构和权重固化下来,再转 TFLite 做量化和部署适配。这样比直接找工具读 .pth稳一些,也方便检查算子有没有不支持。
转之前最好确认模型里有没有自定义算子、动态 shape、复杂后处理这些东西。嵌入式端部署最怕训练时跑得好,转换后一堆算子落不到硬件加速上,最后只能 CPU 跑,性能和功耗都会差不少。
|