From ea90087e38ae6854bcce6a690ea8a4efd97fd6bd Mon Sep 17 00:00:00 2001 From: Jia Guo Date: Fri, 10 Aug 2018 10:13:10 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 298fdf1..ecdf90b 100644 --- a/README.md +++ b/README.md @@ -129,7 +129,7 @@ CUDA_VISIBLE_DEVICES='0,1,2,3' python -u train_softmax.py --network m1 --loss-ty (4). Fine-turn the above Softmax model with Triplet loss. ```Shell -CUDA_VISIBLE_DEVICES='0,1,2,3' python -u train_softmax.py --network m1 --loss-type 12 --lr 0.005 --mom 0.0 --per-batch-size 150 --data-dir ../datasets/faces_ms1m_112x112 --pretrained ../model-m1-softmax,50 --prefix ../model-m1-triplet +CUDA_VISIBLE_DEVICES='0,1,2,3' python -u train_triplet.py --network m1 --lr 0.005 --mom 0.0 --per-batch-size 150 --data-dir ../datasets/faces_ms1m_112x112 --pretrained ../model-m1-softmax,50 --prefix ../model-m1-triplet ``` (5). Train LDPN107E network with Softmax loss on VGGFace2 dataset.