search

harskish / ganspace

Discovering Interpretable GAN Controls [NeurIPS 2020]
Apache License 2.0
1.78k stars 264 forks source link

How to convert .pkl to .pt? #67

Open TopKruZ opened 1 year ago

TopKruZ commented 1 year ago

I have some question, I have my .pkl file and I need to use in ganslace how to convert .pkl file to .pt file?

I try to use convert_weight.py in stylegan2-pytorch but it’s not work , I think because convert_weight.py in stylegan2-pytorch support tensorflow 1.X but I can not install tensorflow 1.X

I follow this step

image0

But “NVlabs/stylegan2” don’t have convert_weight.py , so I download “Stylegan-PyTorch“ instead and run code from picture but….

image1

The report is “No CUDA runtime is found”

I don’t know what does it mean but I installed CUDA toolkits V10.0 and V11.0

Please help me how to solve this issue and how to convert .plk to .pt for use with ganspac.

I try to use pip uninstall torch and use pip install torch is not work.

I try to install tensorflow 1.X version by I have used pip install tensorflow-gpu==1.5 but it’s not work.

‘please help me and teach me how to convert .pkl to .pt file.