PyTorch

共 2 篇文章

本文记录了对 UniMERNet 源码的阅读和理解, 包括类层次结构和四点主要改进.

通过直接找出 ONNX 模型和 PyTorch 模型的权重之间的 Map, 实现 ONNX 模型到 PyTorch 模型的转换.