set weight_x=0 as default

pull/201/head
KaiyangZhou 2019-05-28 23:07:18 +01:00
parent 9963cff629
commit 6dbacc73e6
1 changed files with 2 additions and 2 deletions

View File

@ -139,8 +139,8 @@ def init_parser():
help='number of instances per identity')
parser.add_argument('--weight-t', type=float, default=1,
help='weight to balance hard triplet loss')
parser.add_argument('--weight-x', type=float, default=1,
help='weight to balance cross entropy loss')
parser.add_argument('--weight-x', type=float, default=0,
help='weight to balance cross entropy loss (default is 0)')
# ************************************************************
# Architecture