【pytorch】FutureWarning: You are using torch.load with weights_only=False (the current default value)

错误描述

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 (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-
models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during 
unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via 
`torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. 
Please open an issue on GitHub for any issues related to this experimental feature.

这个警告是关于使用 torch.load() 加载模型时的安全性问题。警告指出，weights_only=False 使用的是 pickle 模块，这可能会执行任意代码，存在潜在的安全隐患。未来版本的 PyTorch 将默认将 weights_only 设置为 True，以限制只加载权重而不加载其他对象。因此，建议显式设置 weights_only=True。

解决方案

修改前

torch.load(BASE_PATH + "/" + self.ModelSelect + '_centroid_matrix.pt')

修改后

torch.load(BASE_PATH + "/" + self.ModelSelect + '_centroid_matrix.pt', weights_only=True)