modelscope如何加载pth文件？

modelscope如何加载pth文件？ makerbi2023-02-24发布于江苏 自然语言处理 训练晚1个epoch后主动停止，得到pth文件，没有output文件夹里的pytorch_model.bin之类的文件，如何处理才能像下面这种用法一样做推理？

text_generation_zh = pipeline(     Tasks.text_generation,      model="./gpt3_poetry_base/output", ) 展开 146 浏览1 关注 4条回答

牧羊吖

月移花影，暗香浮动 可以使用PyTorch的torch.save函数将训练后的模型保存为pytorch_model.bin文件，然后使用BertForSequenceClassification.from_pretrained函数从output/文件夹中加载模型，并将其设置为评估模式。

model = BertForSequenceClassification.from_pretrained("output/") model.eval() 2023-02-24发布于四川 0 1

魏红斌

天下风云出我辈，一入江湖岁月催 我们可以使用模型的load _state_dict方法将它们加载到 resnet18 中。.pth文件报存了模型的权重,这个文件只是一个模型张量参数的pickle文件。

2023-02-24发布于北京 0 0

凌云Cloud

发表文章、提出问题、分享经验、结交志同道合的朋友 你好，我最近也在接触 modelscope 训练模型 ，遇到过类似问题，希望对你有所帮助。

需要将该.pth文件转换为 PyTorch 预训练模型的完整检查点。可以使用convert_gpt2_checkpoint_to_pytorch.py 脚本，然后将生成的文件夹作为模型路径传递给pipeline函数。

python convert_gpt2_checkpoint_to_pytorch.py --model_type=gpt2 --checkpoint_path=path/to/checkpoint.pth --config_file=path/to/config.json --pytorch_dump_folder_path=path/to/pytorch_dump_folder

2023-02-24发布于浙江 1 0

凌云Cloud 转换完成后，得到一个包含完整的检查点文件的文件夹，其中包含pytorch_model.bin 等其他必要的文件，然后此文件夹路径传递给 pipeline 函数即可进行推理。 2023-02-24发布于浙江 0 0

三掌柜666

十分耕耘，一定会有一分收获！ 楼主你好，你可以使用模型的load_state_dict方法将它们加载到resnet18中，因为.pth文件保存了模型的权重，但是该文件只是一个模型张量参数的pickle文件。

2023-02-24发布于江苏 0 1 我到底啦~ + 关注问题 复制 举报 取消

可以使用PyTorch的torch.save函数将训练后的模型保存为pytorch_model.bin文件，然后使用BertForSequenceClassification.from_pretrained函数从output/文件夹中加载模型，并将其设置为评估模式。

model = BertForSequenceClassification.from_pretrained("output/")
model.eval()

我们可以使用模型的load _state_dict方法将它们加载到 resnet18 中。.pth文件报存了模型的权重,这个文件只是一个模型张量参数的pickle文件。

你好，我最近也在接触 modelscope 训练模型 ，遇到过类似问题，希望对你有所帮助。

需要将该.pth文件转换为 PyTorch 预训练模型的完整检查点。可以使用convert_gpt2_checkpoint_to_pytorch.py 脚本，然后将生成的文件夹作为模型路径传递给pipeline函数。

python convert_gpt2_checkpoint_to_pytorch.py --model_type=gpt2 --checkpoint_path=path/to/checkpoint.pth --config_file=path/to/config.json --pytorch_dump_folder_path=path/to/pytorch_dump_folder

楼主你好，你可以使用模型的load_state_dict方法将它们加载到resnet18中，因为.pth文件保存了模型的权重，但是该文件只是一个模型张量参数的pickle文件。