报错原因:模型的网络对不上正在加载的网络,建议检查__init__(),看缺了或多了某些网络结构
RuntimeError:Error(s) in loading state_dict for XXXX: Unexpected key(s) in state_dict
最新推荐文章于 2024-03-06 15:52:52 发布
博客指出Python PyTorch使用时的报错原因,即模型的网络与正在加载的网络不匹配,建议检查__init__(),查看是否存在网络结构缺失或多余的情况。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
2029





