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

move model to gpu in tools/visactmap

pull/294/head
KaiyangZhou 2019-11-27 17:01:32 +00:00
parent bf79ba96be
commit 67f56ca361
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/visualize_actmap.py
View File

@ -134,6 +134,9 @@ def main():
use_gpu=use_gpu use_gpu=use_gpu
) )
if use_gpu:
model = model.cuda()
if args.weights and check_isfile(args.weights): if args.weights and check_isfile(args.weights):
load_pretrained_weights(model, args.weights) load_pretrained_weights(model, args.weights)