Thanks for sharing Convert_pkl_to_pt.ipynb. I do not have a problem with the model provided when run in google colab, but encounter an issue when using my own custom model. My custom model is trained with 512x512 images with stylegan2-ada-pytorch with all default settings.
Traceback (most recent call last):
File "/content/stylegan2-pytorch/convert_weight.py", line 236, in
generator, discriminator, g_ema = pickle.load(f)
ModuleNotFoundError: No module named 'torch_utils'
After I pip install torch_utils, I got the following message:
Traceback (most recent call last):
File "/content/stylegan2-pytorch/closed_form_factorization.py", line 18, in
ckpt = torch.load(args.ckpt)
File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 608, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
ModuleNotFoundError: No module named 'torch_utils.persistence'
Thanks for sharing Convert_pkl_to_pt.ipynb. I do not have a problem with the model provided when run in google colab, but encounter an issue when using my own custom model. My custom model is trained with 512x512 images with stylegan2-ada-pytorch with all default settings.
Traceback (most recent call last): File "/content/stylegan2-pytorch/convert_weight.py", line 236, in generator, discriminator, g_ema = pickle.load(f) ModuleNotFoundError: No module named 'torch_utils'
After I pip install torch_utils, I got the following message: Traceback (most recent call last): File "/content/stylegan2-pytorch/closed_form_factorization.py", line 18, in ckpt = torch.load(args.ckpt) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 608, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 777, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) ModuleNotFoundError: No module named 'torch_utils.persistence'
What should I do?