File size: 670 Bytes
a00ee36
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# 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`