Improved SG2 tf weight conversion instructions

harskish · Dec 11, 2020 · eaa975a · eaa975a · harskish · Dec 11, 2020
1 parent 0ea1051
commit eaa975a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -85,7 +85,7 @@ It is possible to import trained StyleGAN and StyleGAN2 weights from TensorFlow
 2. Modify methods `__init__()`, `load_model()` in `models/wrappers.py` under class StyleGAN.
 
 ### StyleGAN2
-1. Follow the instructions [here](https://github.com/harskish/stylegan2-pytorch/blob/master/README.md#convert-weight-from-official-checkpoints). Make sure to use the forked repository in the conversion for compatibility reasons.
+1. Follow the instructions in [models/stylegan2/stylegan2-pytorch/README.md](https://github.com/harskish/stylegan2-pytorch/blob/master/README.md#convert-weight-from-official-checkpoints). Make sure to use the fork in this specific folder when converting the weights for compatibility reasons.
 2. Save the converted checkpoint as `checkpoints/stylegan2/<dataset>_<resolution>.pt`.
 3. Modify methods `__init__()`, `download_checkpoint()` in `models/wrappers.py` under class StyleGAN2.
 

diff --git a/models/stylegan2/stylegan2-pytorch b/models/stylegan2/stylegan2-pytorch