随笔分类 -  16tensorRT

pth2onnx
摘要：import onnx # 注意这里导入onnx时必须在torch导入之前，否则可能会出现segmentation fault，别人说的 import torch from model_resnet18 import model # 这里是我的模型 device = torch.device("cu 阅读全文
posted @ 2022-07-07 14:23 cheng4632 阅读(427) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示