diff --git a/train_imgreid_xent.py b/train_imgreid_xent.py index fdb8619..84d0b4e 100755 --- a/train_imgreid_xent.py +++ b/train_imgreid_xent.py @@ -35,6 +35,7 @@ def main(): global args torch.manual_seed(args.seed) + np.random.seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() if args.use_cpu: use_gpu = False diff --git a/train_imgreid_xent_htri.py b/train_imgreid_xent_htri.py index e204450..8b911d1 100755 --- a/train_imgreid_xent_htri.py +++ b/train_imgreid_xent_htri.py @@ -36,6 +36,7 @@ def main(): global args torch.manual_seed(args.seed) + np.random.seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() if args.use_cpu: use_gpu = False diff --git a/train_vidreid_xent.py b/train_vidreid_xent.py index a8f80f8..021e9be 100755 --- a/train_vidreid_xent.py +++ b/train_vidreid_xent.py @@ -36,6 +36,7 @@ def main(): global args torch.manual_seed(args.seed) + np.random.seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() if args.use_cpu: use_gpu = False diff --git a/train_vidreid_xent_htri.py b/train_vidreid_xent_htri.py index 11a42a3..dcc5768 100755 --- a/train_vidreid_xent_htri.py +++ b/train_vidreid_xent_htri.py @@ -37,6 +37,7 @@ def main(): global args torch.manual_seed(args.seed) + np.random.seed(args.seed) if not args.use_avai_gpus: os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices use_gpu = torch.cuda.is_available() if args.use_cpu: use_gpu = False