2022年3月27日
【推理引擎】ONNX 模型解析
摘要： 定义模型结构 首先使用 PyTorch 定义一个简单的网络模型： class ConvBnReluBlock(nn.Module): def __init__(self) -> None: super().__init__() self.conv1 = nn.Conv2d(3, 64, 3) sel 阅读全文
posted @ 2022-03-27 10:14 虔诚的树 阅读(3828) 评论(0) 推荐(0) 编辑