pytorch 当尝试转换StyleGan2 Pickle File时，出现pickle.load()错误：'没有名为'torch_utils.persistence'的模块

6ojccjat 于 2023-04-12 发布在其他

关注(0)|答案(1)|浏览(286)

我在一个自定义数据集上训练了一个StyleGan 2-ADA模型，该数据集生成了一个.pkl文件。我现在试图加载.pkl文件，以便将其转换为.pt文件，但当我加载.pkl文件时使用：
pickle.load(f)
我得到一个ModuleNotFoundError: No module named 'torch_utils.persistence'
我已经安装了torch_utils和其他依赖项，但对于加载文件，我不知道如何修复此问题。如果任何人在加载.pkl文件时遇到此问题，任何帮助都将不胜感激！！
Github here上也有同样的问题，但没有明确的解决方案。
已多次尝试安装torch_utils，但错误仍然存在

pytorch

来源：https://stackoverflow.com/questions/75413705/when-trying-to-convert-stylegan2-pickle-file-getting-pickle-load-error-no-m

1条答案

按热度按时间

tktrz96b1#

将dnnlib和torch_utils包包含到您的项目中，您可以在https://github.com/NVlabs/stylegan3中找到它。

import pickle
import torch
import dnnlib
import torch_utils

with open('your.pkl', 'rb') as f:
    G = pickle.load(f)['G_ema'].cpu()  # torch.nn.Module
z = torch.randn([1, G.z_dim]).cpu()  # latent codes
c = None  # class labels (not used in this example)
img = G(z, c)

赞(0）回复(0）举报 2023-04-12

我来回答

pytorch 当尝试转换StyleGan2 Pickle File时，出现pickle.load()错误：'没有名为'torch_utils.persistence'的模块

1条答案

相关问题

热门标签

最新问答