转换StyleGan2 Pickle文件遇pickle.load()错误：找不到torch_utils.persistence模块

以下是几个可尝试的解决方案：

• 使用官方仓库的torch_utils模块
  第三方独立安装的torch_utils包与NVIDIA官方StyleGAN2-ADA仓库中的torch_utils结构不匹配，这是导致找不到persistence子模块的核心原因。请按以下步骤操作：

  1. 克隆NVIDIA官方StyleGAN2-ADA PyTorch仓库到本地
  2. 在加载.pkl文件的脚本开头添加路径引用，确保Python能识别官方的torch_utils模块：

     import sys
     sys.path.insert(0, "/你的本地仓库绝对路径/stylegan2-ada-pytorch")
     

• 清理并检查Python环境

  1. 先卸载可能存在的第三方torch_utils包：

     pip uninstall -y torch_utils
     

  2. 用pip list命令确认当前环境中无第三方torch_utils，再通过上述添加官方仓库路径的方式引入模块。

• 手动构建缺失模块
  若不想克隆整个仓库，可手动创建必要的模块结构：

  1. 在本地创建名为torch_utils的文件夹
  2. 在该文件夹内新建persistence.py文件，复制官方仓库中persistence.py的完整代码
  3. 将这个torch_utils文件夹所在的目录添加到Python路径，方法同第一步的脚本修改。

• 对齐训练与加载环境
  确保加载.pkl文件的环境和训练模型时使用的是同一套StyleGAN2-ADA代码结构，训练时依赖官方仓库代码的话，加载时也必须保持一致，不能混用其他分支或第三方实现。

内容的提问来源于stack exchange，提问作者Naveen M