Skip to content

Commit

Permalink
small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
MasterBin-IIAU committed Jul 24, 2021
1 parent 3a40834 commit 4451bba
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lib/tutorials/STARK_Lightning_Ch.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 4451bba

Please sign in to comment.