rknn-toolkit icon indicating copy to clipboard operation
rknn-toolkit copied to clipboard

Published 20 hours ago • verified publisher icon rockchip-linux

rknn-toolkit一代，pytorch模型是.pth文件可以吗，还是必须.pt的？

Open ttttttt12 opened this issue 2 years ago • 1 comments

rknn-toolkit一代，pytorch模型是.pth文件可以吗，还是必须.pt的？（看手册中写的是后缀.pt的文件

ttttttt12 avatar Jul 28 '22 01:07 ttttttt12

与后缀无关，需要 torch.jit.trace 导出的模型 https://github.com/rockchip-linux/rknn-toolkit/tree/master/examples/pytorch/resnet18

zen-xingle avatar Sep 01 '22 03:09 zen-xingle