Update README.md

This commit is contained in:
chenxuanhong
2022-04-22 02:17:16 +08:00
parent b93603b899
commit 47e5c25351
+2
View File
@@ -67,6 +67,8 @@ The training script is slightly different from the original version, e.g., we re
In order to ensure the normal training, the batch size must be greater than 1.
Friendly reminder, due to the difference in training settings, the user-trained model will have subtle differences in visual effects from the pre-trained model we provide.
- Train 224 models with VGGFace2 224*224 [VGGFace2-224 (10.8G)](https://drive.google.com/file/d/19pWvdEHS-CEG6tW3PdxdtZ5QEymVjImc/view?usp=sharing)
```
python train.py --name simswap224_test --batchSize 4 --gpu_ids 0 --dataset /path/to/VGGFace2HQ --Gdeep False