2023年5月12日
推理加速1_pytorch转tensorRT
摘要： 1.模型转换(pytorch模型->onnx->tensorRT) 1.1 pytorch->onnx dummy_image = torch.randn(1, 3, 224, 224) model.to(device) dummy_image.to(device) dest = "test.onn 阅读全文
posted @ 2023-05-12 10:54 哈哈哈喽喽喽 阅读(78) 评论(0) 推荐(0) 编辑