ZBLOG

FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling...

这个警告是来自PyTorch,提示你在使用 torch.load 加载模型时存在安全风险。具体来说,当 weights_only=False 时,会默认使用 Python 的 pickle 模块,这可能会导致执行不安全的代码,尤其是在加载不可信的数据时。

为了解决这个问题,你可以考虑以下几种方法:

  1. 设置 weights_only=True: 如果你只关心加载模型权重而不需要其他信息,可以将 weights_only 设置为 True。这是推荐的做法,因为它会限制能够被加载的对象,从而提高安全性。
   ckpt = torch.load(file, map_location="cpu", weights_only=True)
  1. 信任文件来源: 如果你完全信任要加载的模型文件,那么可以继续使用当前的方式,但请注意潜在的风险。最好避免从不可信或未知来源下载模型。

  2. 更新 PyTorch 版本: 随着 PyTorch 的更新,该功能可能会进一步完善。确保你的环境中使用的是最新版本的 PyTorch,以便获得最新的安全和功能增强。

  3. 阅读文档: 根据警告中的链接,了解有关如何处理 untrusted models 的更多信息,以及如何使用 torch.serialization.add_safe_globals 来允许特定对象。

总之,为了保证代码运行的安全性,建议尽早适应新的参数设置。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=19128

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?