Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
neuralchen committed Apr 20, 2022
1 parent 8617bea commit 3bf03c8
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,8 @@ If you find this project useful, please star it. It is the greatest appreciation

Download the dataset from [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ).

The training script is slightly different from the original version, e.g., we replace the patch discriminator with the projected discriminator, which saves a lot of hardware overhead and achieves slightly better results.

- Train 256 models
```
python train.py --name simswap256_test --gpu_ids 0 --dataset /path/to/VGGFace2HQ --train_simswap True --Gdeep False
Expand All @@ -73,6 +75,7 @@ python train.py --name simswap512_test --gpu_ids 0 --dataset /path/to/VGGFace2H
```



## Inference with a pretrained SimSwap model
[Preparation](./docs/guidance/preparation.md)

Expand Down

0 comments on commit 3bf03c8

Please sign in to comment.