| # BigGAN-PyTorch TFHub converter |
| This dir contains scripts for taking the [pre-trained generator weights from TFHub](https://tfhub.dev/s?q=biggan) and porting them to BigGAN-Pytorch. |
|
|
| In addition to the base libraries for BigGAN-PyTorch, to run this code you will need: |
|
|
| TensorFlow |
| TFHub |
| parse |
|
|
| Note that this code is only presently set up to run the ported models without truncation--you'll need to accumulate standing stats at each truncation level yourself if you wish to employ it. |
|
|
| To port the 128x128 model from tfhub, produce a pretrained weights .pth file, and generate samples using all your GPUs, run |
|
|
| `python converter.py -r 128 --generate_samples --parallel` |