From 8112ca15491ba701527f90365c2d2409cb584729 Mon Sep 17 00:00:00 2001 From: chenxuanhong Date: Fri, 22 Apr 2022 00:50:05 +0800 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 3ac227a..8338643 100644 --- a/README.md +++ b/README.md @@ -70,6 +70,9 @@ In order to ensure normal training, the batch size must be greater than 1. python train.py --name simswap224_test --batchSize 4 --gpu_ids 0 --dataset /path/to/VGGFace2HQ --Gdeep False ``` +[Colab demo for training 224 model](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb) + + - Train 512 models with VGGFace2-HQ 512*512 [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ). ``` python train.py --name simswap512_test --gpu_ids 0 --dataset /path/to/VGGFace2HQ --Gdeep True