jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix potential oom issue (#387)

Browse Source
This commit is contained in:
cdpath 2023-09-12 10:41:03 +08:00 committed by GitHub
parent b9b145c335
commit 722eb39526
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
opencompass/openicl/icl_retriever/icl_mdl_retriever.py
View File

@ -141,6 +141,7 @@ class MDLRetriever(TopkRetriever):
"""Retrieve the in-context example index for each test example.""" """Retrieve the in-context example index for each test example."""
return self.topk_search() return self.topk_search()
@torch.no_grad()
def cal_ce(self, input_texts: List[str], mask_length=None): def cal_ce(self, input_texts: List[str], mask_length=None):
if self.metric_model is None: if self.metric_model is None:
logger.info( logger.info(