摘要: pytorch 转onnx 首先加载pytorch模型 # load model import torch def load_model(ckpt) # build model model = build_model() # depending on your own model build fun 阅读全文
posted @ 2022-05-06 15:31 老张哈哈哈 阅读(2270) 评论(0) 推荐(1) 编辑