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] keep keys (#431)

Browse Source
This commit is contained in:
Tong Gao 2023-09-22 17:30:54 +08:00 committed by GitHub
parent a1ea3c094a
commit 07574fddbb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/partitioners/base.py
View File

@ -62,7 +62,7 @@ class BasePartitioner:
tgt_ptr[key] = {} tgt_ptr[key] = {}
tgt_ptr = tgt_ptr[key] tgt_ptr = tgt_ptr[key]
tgt_ptr[key_chain[-1]] = ori_ptr[key_chain[-1]] tgt_ptr[key_chain[-1]] = ori_ptr[key_chain[-1]]
except AttributeError: except Exception:
self.logger.warning(f'Key {k} not found in config, ignored.') self.logger.warning(f'Key {k} not found in config, ignored.')
tasks = self.partition(models, tasks = self.partition(models,