摘要: 模型转换为onnx import torch import torchvision dummy_input = torch.randn(10, 3, 224, 224, device="cuda") model = torchvision.models.alexnet(pretrained=True 阅读全文
posted @ 2020-12-04 10:44 早上好章鱼哥 阅读(2866) 评论(0) 推荐(0) 编辑