faiss升级numpy版本惹的祸
2020-10-08 本文已影响0人
城中山谷
最近在进行Facebook的基于faiss语义召回用于电商搜索的召回开发工作时候,基于tensorflow在训练dssm模型的过程中,模型的代码在图的run过程中,出现了莫名其妙的bug,信息如下:
TypeErrorTraceback (most recent call last)TypeError: int() argument must be a string, a bytes-like object or a number, not 'list'The above exception was the direct cause of the following exception:ValueErrorTraceback (most recent call last)
....
ValueError: setting an array element with a sequence.
困扰很久,原因是:安装faiss时候把numpy版本升级为了1.19.1原来的1.15.1被替换了;但是把版本回退到1.15.1时候faiss计算向量相似性时候又出错,所以只能分开两个虚拟环境分别进行两个阶段的工作,具体原因还未查看源码。
参考资料:
https://www.liangzl.com/get-article-detail-202136.html
https://stackoverflow.com/questions/63097829/debugging-numpy-visibledeprecationwarning-ndarray-from-ragged-nested-sequences