zc-maker
zc-maker
@Amanda-Barbara ,您好,如何转换为pytorch之后在转换为rknn呢?
@Amanda-Barbara 谢谢指点,您如何转换paddle模型为pytorch的pt格式呢?最近也在尝试做这项工作,目前采用的是onnx直接进行部署的,纯cpu跑的,方便的话,加个vx,我想请教一下您 我的:601633950
@Amanda-Barbara 谢谢您,我知道这个项目,在保存jit时我遇到了一些问题 model=torch.jit.trace(converter.net,inp,strict=False) model(inp) torch.jit.save(model, "det.pt") 采用这样的方式保存,用rknn加载会报错,您有jit保存的建议吗?我很感谢您的回复
@Amanda-Barbara 谢谢指点,但是我尝试过这样的保存方式,但是在进行保存的时候,报错了,必须要加strict=False这个参数,但是这样保存的之后的模型无法再rknn中进行加载 RuntimeError: Encountering a dict at the output of the tracer might cause the trace to be incorrect, this is only valid if the container structure does not change...
@Amanda-Barbara 好的,非常感谢,我再试试,您说的paddleocr2pytorch的convert脚本要和paddleocr模型的网络结构要对应的上,是要修改转换脚本吗?
@Amanda-Barbara https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/doc/doc_ch/models_list.md 您转换的是这个连接下的det模型吗?我测试都会有这个警告
I have a same error : when I train use loss.backward() llvm::SmallVector mlir::triton::gpu::getElemsPerThread(mlir::Attribute, llvm::ArrayRef, mlir::Type): Assertion `0 && "getElemsPerThread not implemented"' failed.