Pytorch版本问题
将
param_dict = torch.load(trained_path)
改为
param_dict = torch.load(trained_path).state_dict()
即可
修复Pytorch加载模型参数问题
博客内容涉及Pytorch模型加载时出现的问题及其解决方案。通过将`torch.load(trained_path)`更改为`torch.load(trained_path).state_dict()`,可以成功加载模型的参数状态。
Pytorch版本问题
将
param_dict = torch.load(trained_path)
改为
param_dict = torch.load(trained_path).state_dict()
即可
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
6024

被折叠的 条评论
为什么被折叠?