摘要: 一、概述 首先,我们导入需要的包,包括模型,优化器,梯度求导等,将数据类型全部转化成tensor类型 import numpy as np import torch import torch.nn as nn import torch.optim as optim from torch.autogr 阅读全文
posted @ 2021-05-21 15:17 jasonzhangxianrong 阅读(492) 评论(0) 推荐(0) 编辑
摘要: Faiss库是由 Facebook 开发的适用于稠密向量匹配的开源库,支持 c++ 与 python 调用。 通过实验证实,128维的125W向量,在 CPU 下检索耗时约70ms,经过 GPU 加速后检索耗时仅5ms。 一、安装 Faiss 支持直接通过 conda 安装 python 接口,以及 阅读全文
posted @ 2021-05-21 11:39 jasonzhangxianrong 阅读(3154) 评论(0) 推荐(0) 编辑