5f465041619a63bf1de99c55d1298865bf70fd9e,reconstruction.py,,,#,19

Before Change

import timeit


MANIFOLDS = {
    "lorentz": LorentzManifold,
    "euclidean": EuclideanManifold,
    "transe": TranseManifold,
    "poincare": PoincareManifold
}

np.random.seed(42)

parser = argparse.ArgumentParser()

After Change

model = MODELS[chkpnt["conf"]["model"]](
    manifold,
    dim=chkpnt["conf"]["dim"],
    size=chkpnt["embeddings"].size(0),
    sparse=chkpnt["conf"]["sparse"]
)
model.load_state_dict(chkpnt["model"])

Instances