开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope这个ckpt怎么转成bin文件?

ModelScope这个ckpt怎么转成bin文件?image.png

展开
收起
鸡蛋灌饼儿 2023-02-19 14:58:19 432 0
1 条回答
写回答
取消 提交回答
  • 可以使用PyTorch的torch.save()函数来实现这个目的。参考下面示例:

    python下,

    import torch
    from modelscope import get_model

    获取模型

    model = get_model('path/to/model.pt')

    将模型保存为bin文件

    torch.save(model.state_dict(), 'path/to/model.bin')

    2023-05-23 19:30:38
    赞同 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载