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

rm .cpu() in saving models

pull/17/head
KaiyangZhou 2018-03-29 16:45:39 +01:00
parent 6c6bd95e33
commit af853bbbaf
5 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_img_model_cent.py
View File

@ -166,7 +166,7 @@ def main():
if is_best: best_rank1 = rank1
if use_gpu:
state_dict = model.module.cpu().state_dict()
state_dict = model.module.state_dict()
else:
state_dict = model.state_dict()
save_checkpoint({

2
train_img_model_xent.py
View File

@ -159,7 +159,7 @@ def main():
if is_best: best_rank1 = rank1
if use_gpu:
state_dict = model.module.cpu().state_dict()
state_dict = model.module.state_dict()
else:
state_dict = model.state_dict()
save_checkpoint({

2
train_img_model_xent_htri.py
View File

@ -168,7 +168,7 @@ def main():
if is_best: best_rank1 = rank1
if use_gpu:
state_dict = model.module.cpu().state_dict()
state_dict = model.module.state_dict()
else:
state_dict = model.state_dict()
save_checkpoint({

2
train_vid_model_xent.py
View File

@ -167,7 +167,7 @@ def main():
if is_best: best_rank1 = rank1
if use_gpu:
state_dict = model.module.cpu().state_dict()
state_dict = model.module.state_dict()
else:
state_dict = model.state_dict()
save_checkpoint({

2
train_vid_model_xent_htri.py
View File

@ -176,7 +176,7 @@ def main():
if is_best: best_rank1 = rank1
if use_gpu:
state_dict = model.module.cpu().state_dict()
state_dict = model.module.state_dict()
else:
state_dict = model.state_dict()
save_checkpoint({