Skip to content
Snippets Groups Projects
Commit 9a32cc27 authored by lenhy's avatar lenhy
Browse files

Bugfix

parent 29060a91
Branches master
No related tags found
No related merge requests found
......@@ -45,8 +45,8 @@ def get_model(name: str, triples, inverse_triples, regularizer):
"ConvE": 0.1,
"DistMult": 0.0,
}
model = model_dict[name]
if name == "TransE":
model = model_dict[name]
model.loss.margin = 2.0
return model, lr_dict[name], epochs_dict[name], bs_dict[name], ls_dict[name]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment