将PKL (Stylegan2-ada)转换为PT (Pytorch模型)时失败

Question

我有个问题。

当我尝试将我的pkl模型转换成pt模型时，我得到了这个错误。

  File "/content/stylegan2-pytorch/convert_weight.py", line 221, in <module>
    generator, discriminator, g_ema = pickle.load(f)
  File "./stylegan2/dnnlib/tflib/network.py", line 278, in __setstate__
    assert state["version"] in [2, 3, 4]
AssertionError

我不知道该怎么做，也不知道如何成功地转换我的模型。有人能帮帮我吗？我真的很感激！

Answer 1

最新的network.py文件包含了更多的state["version"] ([2, 3, 4, 5])，你只需要找到新版本的stylegan2-ada即可。