From 9726b8bb960504eab9f8e4d2a0654268d8060de2 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Mon, 5 Nov 2018 21:25:09 +0000 Subject: [PATCH] set required=True for dataset arguments --- torchreid/data_manager.py | 2 +- train_imgreid_xent.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/torchreid/data_manager.py b/torchreid/data_manager.py index 5497cfc..b406748 100644 --- a/torchreid/data_manager.py +++ b/torchreid/data_manager.py @@ -14,7 +14,7 @@ class ImageDataManager(object): workers, pin_memory, **kwargs): self.train_names = train_names - self.test_names = test_names if test_names is not None else train_names + self.test_names = test_names self.train = [] self.num_train_pids = 0 diff --git a/train_imgreid_xent.py b/train_imgreid_xent.py index 78d37ac..ab8fbd9 100755 --- a/train_imgreid_xent.py +++ b/train_imgreid_xent.py @@ -31,8 +31,8 @@ parser = argparse.ArgumentParser(description='Train image model with cross entro # Datasets parser.add_argument('--root', type=str, default='data', help="root path to data directory") -parser.add_argument('-s', '--source', type=str, default=['market1501'], nargs='+') -parser.add_argument('-t', '--target', type=str, nargs='+') +parser.add_argument('-s', '--source', type=str, required=True, nargs='+') +parser.add_argument('-t', '--target', type=str, required=True, nargs='+') parser.add_argument('-j', '--workers', default=4, type=int, help="number of data loading workers (default: 4)") parser.add_argument('--height', type=int, default=256,