RK3399Pro的问题记录
最新推荐文章于 2024-09-24 10:32:19 发布
博客讲述了在使用PyTorch模型遇到的问题,即普通PyTorch模型无法在RK3399Pro上加载和转换。通过分析SDK示例代码,作者发现需要使用torch.jit.trace()将模型转换为标准格式才能进行RKNN的加载。提供了一个torch2jit.py的代码片段来展示如何实现这一转换过程。
摘要由CSDN通过智能技术生成