The models in networks_tf.py
can be used with the weights from the official repository, which I have converted to PyTorch state dicts.
Download converted weights: Places2 | CelebA-HQ (for networks_tf.py
)
The networks in networks_tf.py
use TensorFlow-compatibility functions (padding, down-sampling), while the networks in networks.py
do not. In order to adjust the weights to the different settings, the model was trained on Places2/CelebA-HQ for some time using the pretrained weights as initialization.
Download fine-tuned weights: Places2 | CelebA-HQ (for networks.py
)