tensorrt加速pytorch生成pth: pth->onnx->engine
生成静态engine模型--batch=1一、pytorch模型保存1、保存模型参数save_filename = 'net_%s.pth'% epoch_labelsave_path = os.path.join('./model',name,save_filename)torch.save(network.cpu().state_dict(), save_path)导入模型参数save_path = os.path.join('./model',name,'net_%s.pt
原创
2021-03-15 10:52:51 ·
2355 阅读 ·
3 评论