开发者社区 > ModelScope模型即服务 > 自然语言处理 > 正文

重新训练的pth模型文件,怎么转成bin文件?

已解决

重新训练的pth模型文件,怎么转成bin文件?因为我看到模型文件是pytorch_model.bin的

展开
收起
哈喽!小陈 2023-02-24 09:47:55 1200 0
1 条回答
写回答
取消 提交回答
  • 推荐回答

    import torch import sys

    #pth文件路径 model_path = sys.argv[1] #bin文件路径 out_path = sys.argv[2]

    checkpoint = torch.load(model_path, map_location='cpu') state_dict = checkpoint['state_dict'] torch.save(state_dict, out_path)

    2023-02-24 10:25:02
    赞同 展开评论 打赏

包含命名实体识别、文本分类、分词、关系抽取、问答、推理、文本摘要、情感分析、机器翻译等多个领域

热门讨论

热门文章

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载