请教下,我尝试将 cv_mobilenet-v2_bad-image-detecting 这个模型转

请教下,我尝试将 cv_mobilenet-v2_bad-image-detecting 这个模型转换成onnx格式,在你们提供的cpu镜像容器中,执行

model = torch.load('pytorch_model.pt') model.eval()

提示 : 'collections.OrderedDict' object has no attribute 'eval'

针对你们的模型,有没有可用的 pytorch转onnx的脚本?

展开
收起
真的很搞笑 2023-04-22 14:23:25 335 分享 版权
1 条回答
写回答
取消 提交回答
  • 不同的模型转onnx需要定制化的脚本 ,我找到了模型的负责同学问了下,这个模型暂时不支持onnx导出。另外,您对模型的使用方式是有问题的,可以参考https://www.modelscope.cn/models/damo/cv_mobilenet-v2_bad-image-detecting/summary 中的代码范例进行推理,如果需要初始化模型可以:Model.from_pretrained('damo/cv_mobilenet-v2_bad-image-detecting'),此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-04-27 08:11:33
    赞同 展开评论

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

还有其他疑问?
咨询AI助理