2020年12月21日
onnx算子大全
摘要： 本文通过此脚本从def文件自动生成。不要直接修改，而是编辑算子定义。 对于算子输入/输出的可辩别的，它可以是可辩别的、不可辩别的或未定义的。如果未指定变量的可辩别的，则该变量具有未定义的可辩别的。 ai.onnx (default) OperatorSince version Abs 13, 6, 阅读全文
posted @ 2020-12-21 08:26 吴建明wujianming 阅读(7482) 评论(0) 推荐(2) 编辑
从PyTorch到ONNX的端到端AlexNet
摘要： 从PyTorch到ONNX的端到端AlexNet 这是一个简单的脚本，可将Torchvision中定义的经过预训练的AlexNet导出到ONNX中。运行一轮推理Inference，然后将生成的跟踪模型保存到alexnet.onnx： import torch import torchvision d 阅读全文
posted @ 2020-12-21 07:44 吴建明wujianming 阅读(337) 评论(0) 推荐(0) 编辑
使用TENSORRT和NVIDIA-DOCKER部署深部神经网络
摘要： 使用TENSORRT和NVIDIA-DOCKER部署深部神经网络 当前部署工作流 阅读全文
posted @ 2020-12-21 07:07 吴建明wujianming 阅读(662) 评论(0) 推荐(0) 编辑
用于ONNX的TensorRT后端
摘要： 用于ONNX的TensorRT后端 解析ONNX模型以使用TensorRT执行。 另请参阅TensorRT文档。 有关最近更改的列表，请参见changelog。 支持的TensorRT版本 Master分支上的开发适用于具有完整维度和动态架构shape支持的TensorRT 7.2.1的最新版本。 阅读全文
posted @ 2020-12-21 06:23 吴建明wujianming 阅读(1431) 评论(0) 推荐(0) 编辑