Skip to content

Where is the step 'agent selection' and 'team reformation' in llmlp_listwise_mmlu.py #2

@ME-Msc

Description

@ME-Msc

for que, ans in qa_pairs:
llmlp.zero_grad()
res, resp_cnt, completions, prompt_tokens, completion_tokens = llmlp.forward(que)
imp_score = llmlp.backward(res)
completion_list.append(completions)
accs.append(ans == res)
resp_cnts += resp_cnt
importances.append(imp_score)
total_prompt_tokens += prompt_tokens
total_completion_tokens += completion_tokens
with open(DIR_NAME+'/'+EXP_NAME+'_'+str(len(ROLES))+'3.json', 'a') as f:
f.write(json.dumps(completions) + '\n')

As the title describes, I didn't see the step 'agent selection' and 'team reformation' in llmlp_listwise_mmlu.py lines 45-58. You only did forward (propagation in section 3.4(1) of your paper) and backward (aggregation in section 3.4(2) of your paper).

Could you update the code? Thanks.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions