From 4451bba649b888c2f9ce1f467759e485c4c85ea9 Mon Sep 17 00:00:00 2001 From: Bin Yan <40926230+MasterBin-IIAU@users.noreply.github.com> Date: Sat, 24 Jul 2021 19:35:39 +0800 Subject: [PATCH] small fix --- lib/tutorials/STARK_Lightning_Ch.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/tutorials/STARK_Lightning_Ch.md b/lib/tutorials/STARK_Lightning_Ch.md index 357fc5c..31a37c7 100644 --- a/lib/tutorials/STARK_Lightning_Ch.md +++ b/lib/tutorials/STARK_Lightning_Ch.md @@ -13,7 +13,7 @@ pip install onnx onnxruntime-gpu==1.6.0 ``` pip install onnx onnxruntime ``` -##ONNX模型转换与推理测试 +## ONNX模型转换与推理测试 下载训练好的PyTorch模型权重文件 [STARK_Lightning](https://drive.google.com/file/d/18xxbMKCjWi6Gvn5T4o2w5jIbwd3AWN55/view?usp=sharing) 将训练好的PyTorch模型转换成onnx格式,并测试onnxruntime