mirrors
/
PyRetri
mirror of https://github.com/PyRetri/PyRetri.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

convert to cpu()

master
hubenyi 2020-06-29 17:03:21 +08:00
parent b2e189d483
commit 53bd3b3768
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main/single_index.py
View File

@ -50,7 +50,7 @@ def main():
stacked_feature = list()
for name in cfg.index.feature_names:
assert name in img_fea_info[0], "invalid feature name: {} not in {}!".format(name, img_fea_info[0].keys())
stacked_feature.append(img_fea_info[0][name])
stacked_feature.append(img_fea_info[0][name].cpu())
img_fea = np.concatenate(stacked_feature, axis=1)
# load gallery features