From 0374e33c7c449dcb99463aa591f173d381e308c9 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Tue, 23 Oct 2018 11:37:03 +0100 Subject: [PATCH] add 'global' keyword to args, best_rank1 --- train_imgreid_xent.py | 2 ++ train_imgreid_xent_htri.py | 2 ++ train_vidreid_xent.py | 2 ++ train_vidreid_xent_htri.py | 2 ++ 4 files changed, 8 insertions(+) diff --git a/train_imgreid_xent.py b/train_imgreid_xent.py index 3ca265a..150a327 100755 --- a/train_imgreid_xent.py +++ b/train_imgreid_xent.py @@ -109,6 +109,8 @@ best_rank1 = -np.inf def main(): + global args, best_rank1 + torch.manual_seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() diff --git a/train_imgreid_xent_htri.py b/train_imgreid_xent_htri.py index 99755be..e4e4a54 100755 --- a/train_imgreid_xent_htri.py +++ b/train_imgreid_xent_htri.py @@ -114,6 +114,8 @@ best_rank1 = -np.inf def main(): + global args, best_rank1 + torch.manual_seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() diff --git a/train_vidreid_xent.py b/train_vidreid_xent.py index 9bf01b5..97ab019 100755 --- a/train_vidreid_xent.py +++ b/train_vidreid_xent.py @@ -103,6 +103,8 @@ best_rank1 = -np.inf def main(): + global args, best_rank1 + torch.manual_seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() diff --git a/train_vidreid_xent_htri.py b/train_vidreid_xent_htri.py index 7dbcf4d..c3472d7 100755 --- a/train_vidreid_xent_htri.py +++ b/train_vidreid_xent_htri.py @@ -108,6 +108,8 @@ best_rank1 = -np.inf def main(): + global args, best_rank1 + torch.manual_seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available()