ModelScope的pipeline加载自己训练的模型时出了问题怎么办?

ModelScope的pipeline加载自己训练的模型时出了问题怎么办?

展开
收起
青城山下庄文杰 2023-08-12 09:45:19 252 分享 版权
1 条回答
写回答
取消 提交回答
  • 您试试下面方法:
    pipeline里的model='xxx' # xxx为保存训练时设置的work_dir在pipeline下加入下面代码

    checkpoint = load_checkpoint(
                        p.model,
                        '/x/xxx/xxxx/xxxxx/best_accuracy_top-1_epoch_1.pth', # work_dir下的best开头的.pth文件
                        map_location='cpu')
    if 'CLASSES' in checkpoint.get('meta', {}):
        CLASSES = checkpoint['meta']['CLASSES']
    p.model.CLASSES=CLASSES
    
    # 后面使用pipeline推理
    p('img_path')
    

    -此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-08-15 08:22:59
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

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