Skip to content

Commit

Permalink
Merge branch 'develop' of https://github.com/dice-group/dice-embeddings
Browse files Browse the repository at this point in the history
… into checkpoint-disabling
  • Loading branch information
sshivam95 committed Nov 30, 2024
2 parents b6294b6 + 787b535 commit e7a86b4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dicee/static_funcs_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def make_iterable_verbose(iterable_object, verbose, desc="Default", position=Non
else:
return iterable_object


@torch.no_grad()
def evaluate_lp(model, triple_idx, num_entities, er_vocab: Dict[Tuple, List], re_vocab: Dict[Tuple, List],
info='Eval Starts', batch_size=128, chunk_size=1000):
"""
Expand Down

0 comments on commit e7a86b4

Please sign in to comment.