search

odegeasslbc / FastGAN-pytorch

Official implementation of the paper "Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis" in ICLR 2021
GNU General Public License v3.0
600 stars 100 forks source link

.pkl instead of .pth #6

Closed corlangerak closed 3 years ago

corlangerak commented 3 years ago

How could FastGAN generate pickle .pkl files instead of .pth files. Wouldn't there be more compatibility with stylegan2-ada? Or am I missing something? Could you convert this. or change in manually in the code?

odegeasslbc commented 3 years ago

Hi

You can use "torch.load()" to load the saved .pth file, which does not have any compatibility issue within the PyTorch environment. Actually '.pth' is not a file format, I just use it to represent files that are saved model weights.

I believe the "stylegan2-ada" you refer to is the official implementation, which uses TensorFlow, a different framework than PyTorch.