yolov5我不知道,但是我遇到类似问题,在移植arcface的时候,模型是高版本pytorch.重新加载再保存模型,设置use_new_zipfile_serialization=False。然后再转cambricon.
代码如下:
weight_cp =weight.replace(".pth","_no_use_new_zip.pth")
torch.save(model.state_dict(), model_cp,_use_new_zipfile_serialization=False) # 训练所有数据后,保存网络的参数
请登录后评论