ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-19 15:41:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

triplet fix

Browse Source
This commit is contained in:
nttstar
2017-12-26 12:30:19 +08:00
parent 8f5461d74d
commit a4d2910742
2 changed files with 64 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/train_softmax.py
View File

@@ -484,7 +484,7 @@ def train_net(args):
label_names = (label_name,),
)
if len(data_dir_list)==1:
if len(data_dir_list)==1 and args.loss_type!=12:
train_dataiter = FaceImageIter(
batch_size = args.batch_size,
data_shape = data_shape,