2022年10月27日
PyTorch 1.0 中文文档：torch.onnx
摘要： 译者：guobaoyo 示例:从Pytorch到Caffe2的端对端AlexNet模型 这里是一个简单的脚本程序,它将一个在 torchvision 中已经定义的预训练 AlexNet 模型导出到 ONNX 格式. 它会运行一次,然后把模型保存至 alexnet.onnx: import torch 阅读全文
posted @ 2022-10-27 22:39 绝不原创的飞龙 阅读(57) 评论(0) 推荐(0)