Open smilepolker opened 5 months ago
根据报错,你可能没有安装rknn toolkit2 的PC的WHEEL
额,我应该是安装了的,我这边的环境时使用cpu进行训练,不确定是否和这个环境有关,但是现在已经解决这个问题了,我对保存MobileNetV3.pt的代码进行了修改,然后就可以正常转换了。 修改的文件是train.py, 代码行数是:95
input_tensor = torch.rand(1, 3, 224, 224)
traced_script_module = torch.jit.trace(model, input_tensor)
# save
traced_script_module.save("MobileNetV3.pt")
# torch.save(model, "MobileNetV3.pt")
你能你能否帮忙检查下我的代码是否有问题,然后将你的代码仓库更新下
问题描述:在执行训练完后生成的MobileNetV3.pt无法通过pt2rknn.py转换为tspi_moblienetv3_demo.rknn 版本:当前版本 复现步骤: