Class DIST

knn_utils/saveEmbedMulti.py:25–34

import faiss.contrib.torch_utils

class DIST(Enum):
    l2 = auto()
    dot = auto()

    @staticmethod
    def from_string(s):
        try:
            return DIST[s.lower()]
        except KeyError:
            raise ValueError()

class KEY_TYPE(Enum):
    last_ffn_input = auto()