weixin_68258311的博客

原创 YOLOv5训练报错 UnpicklingError: Weights only load failed.

ckpt = torch.load(weights, map_location=device, weights_only=False) # 添加安全参数。正常配置环境之后运行train.py报错如上，从错误信息来看，这个问题是由于PyTorch 2.6版本对权重加载的安全限制导致的。