starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix code

Browse Source
This commit is contained in:
qinxy3
2023-08-15 11:03:24 +08:00
parent bff91b3305
commit 543ffaf617
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
eval/evaluate_ceval.py
View File

@@ -95,10 +95,10 @@ def eval_subject(
softval = torch.nn.functional.softmax( softval = torch.nn.functional.softmax(
torch.tensor( torch.tensor(
[ [
logits[tokenizer("A")['input_ids'][-1]], logits[tokenizer("A")['input_ids']],
logits[tokenizer("B")['input_ids'][-1]], logits[tokenizer("B")['input_ids']],
logits[tokenizer("C")['input_ids'][-1]], logits[tokenizer("C")['input_ids']],
logits[tokenizer("D")['input_ids'][-1]], logits[tokenizer("D")['input_ids']],
] ]
), ),
dim=0, dim=0,