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

已解决

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

展开
收起
哈喽!小陈 2023-02-24 09:47:55 1315 分享 版权
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
    赞同 展开评论

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

热门讨论

热门文章

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