torch.load()失败/错误的原因总结

Traceback (most recent call last): File "C:\Users\asus\Desktop\Yolo+realsense实时获取目标物三维坐标代码\interface.py", line 12, in <module> model = torch.hub.load('./', 'custom', 'small.pt', source='local') File "C:\Anaconda\envs\myenv\lib\site-packages\torch\hub.py", line 647, in load model = _load_local(repo_or_dir, model, *args, **kwargs) File "C:\Anaconda\envs\myenv\lib\site-packages\torch\hub.py", line 676, in _load_local model = entry(*args, **kwargs) File "C:\Users\asus\Desktop\Yolo+realsense实时获取目标物三维坐标代码\./hubconf.py", line 69, in custom model = torch.load(path_or_model) if isinstance(path_or_model, str) else path_or_model # load checkpoint File "C:\Anaconda\envs\myenv\lib\site-packages\torch\serialization.py", line 1470, in load raise pickle.UnpicklingError(_get_wo_message(str(e))) from None _pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, do those steps only if you trust the source of the checkpoint. (1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source. (2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message. WeightsUnpickler error: Unsupported global: GLOBAL numpy.core.multiarray._reconstruct was not an allowed global by default. Please use `torch.serialization.add_safe_globals([_reconstruct])` or the `torch.serialization.safe_globals([_reconstruct])` context manager to allowlist this global if you trust this class/function. Check the documentation of torch.load to learn more about types accepted by default with weights_only https://pytorch.org/docs/stable/generated/torch.load.html. requirements: C:\Users\asus\Desktop\Yolo+realsense实时获取目标物三维坐标代码\requirements.txt not
03-29
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值