Update inference.py

Yukun-Huang · Jun 3, 2020 · 1f720b5 · 1f720b5
1 parent 3e95b4c
commit 1f720b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference.py b/inference.py
@@ -37,7 +37,7 @@
 assert args.dataset in ['market', 'duke']
 assert args.backbone in ['resnet50', 'resnet34', 'resnet18', 'densenet121']
 
-model_name = '{}_nfc_softmax'.format(args.backbone) if args.use_id else '{}_nfc'.format(args.backbone)
+model_name = '{}_nfc_id'.format(args.backbone) if args.use_id else '{}_nfc'.format(args.backbone)
 num_label, num_id = num_cls_dict[args.dataset], num_ids_dict[args.dataset]