mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add np.random.seed

pull/119/head
KaiyangZhou 2019-01-09 21:15:21 +00:00
parent 10b074e9b7
commit 1dce953786
4 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_imgreid_xent.py
View File

@ -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

1
train_imgreid_xent_htri.py
View File

@ -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

1
train_vidreid_xent.py
View File

@ -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

1
train_vidreid_xent_htri.py
View File

@ -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