search

huggingface / pytorch-pretrained-BigGAN

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.
MIT License
1.02k stars 180 forks source link

How to load the model?? #2

Open xiaosuzhang opened 5 years ago

xiaosuzhang commented 5 years ago

model = BigGAN.from_pretrained('biggan-deep-256') Can we just replace 'biggan-deep-256' with the path which we download the pre-train model??

thomwolf commented 5 years ago

Hi, which pre-trained model are you referring to?

xiaosuzhang commented 5 years ago

I download the pre-trained model named“biggan-deep-256-pytorch_model.bin”