From 6638da6e7e5e55b8f492050843447ee64a62cfc6 Mon Sep 17 00:00:00 2001
From: Shivam Sharma <sshivam95@gmail.com>
Date: Sat, 30 Nov 2024 15:53:03 +0100
Subject: [PATCH] Fix memory allocation issue

Issue #284
---
 dicee/static_funcs_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dicee/static_funcs_training.py b/dicee/static_funcs_training.py
index 6e08797c..e92a5d43 100644
--- a/dicee/static_funcs_training.py
+++ b/dicee/static_funcs_training.py
@@ -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):
     """