2023年8月28日
简单的将pytorch模型部署到onnx
摘要： 1. 创建一个pytorch模型 这里我用的U2Net，直接加载好训练出的权重 model = U2Net(class_nums=4) model.load_state_dict(torch.load(checkpoint_path)) 2. 将pytorch模型转成onnx格式 x = torcg 阅读全文
posted @ 2023-08-28 10:12 Liang-ml 阅读(358) 评论(0) 推荐(0)