Function main

src/opensource_functioncall_evalution.py:196–218

()

def main():
    pool = ThreadPoolExecutor(max_workers=GPU_NUM)

    test_lines = []
    for test_ans_file in test_ans_file_list:
        print(test_ans_file)
        with open(test_ans_file, "r") as f:
            lines = f.readlines()
            test_lines += lines

    batch_num = len(test_lines)//GPU_NUM + int(len(test_lines)%GPU_NUM>0)

    obj_list = []
    for idx in range(GPU_NUM):
        batch_test_lines = test_lines[idx*batch_num:(idx+1)*batch_num]
        obj = pool.submit(test_process, batch_test_lines, gpu_index=idx)
        obj_list.append(obj)

    for future in as_completed(obj_list):
        # 暂时留在这里，但是其实没有返回数据
        data = future.result()

    print_result()

if __name__ == "__main__":
    main()