摘要: 使用torch.onnx.export来进行模型的构造 import torch import torch.nn as nn import torch.nn.functional as F import torch.onnx import netron class model(nn.Module): 阅读全文
posted @ 2021-08-18 22:48 python我的最爱 阅读(2264) 评论(0) 推荐(0) 编辑