更新(20200429)
此存储库基于库。 将这个高效的付款网络转换为ONNX的需求很多,因此我们进行此回购以帮助人们将模型转换为ONNX或TVM。 请注意,此存储库仅提供如何将模型转换为ONNX或TVM的功能,而不关注模型训练或其他事项。 如果您想训练或测试这种有效模型,最好的方法是参考原始的回购。
我们已根据此更改了一些代码,以帮助成功进行转换。
转换onnx
如果要转换为ONNX,只需运行
python3 convert/convert_onnx.py
转换电视
我们已经在提交** f08d5d78ee000b2c113ac451f8d73817960eafd5 **的tvm版本上进行了测试,其他版本未经测试,因此无法确保也能正常工作。 首先,您需要安装tvm,请参阅其。 我们称您的tvm安装源目录为tvm_home 。 然后,在tvm_home/python/tvm/rela
2021-07-04 14:33:21
8.8MB
Python
1