tangger/Search-R1
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix action status

Browse Source
This commit is contained in:
PeterGriffinJin
2025-03-22 14:49:23 +00:00
parent 4936a3115e
commit 6272082a64
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
search_r1/llm_agent/generation.py
View File

@@ -307,16 +307,17 @@ class LLMGenerationManager:
valid_action_stats += torch.tensor(valid_action, dtype=torch.int) valid_action_stats += torch.tensor(valid_action, dtype=torch.int)
valid_search_stats += torch.tensor(is_search, dtype=torch.int) valid_search_stats += torch.tensor(is_search, dtype=torch.int)
meta_info['turns_stats'] = turns_stats.tolist()
meta_info['active_mask'] = active_mask.tolist()
meta_info['valid_action_stats'] = valid_action_stats.tolist()
meta_info['valid_search_stats'] = valid_search_stats.tolist()
original_right_side = self._update_right_side( original_right_side = self._update_right_side(
original_right_side, original_right_side,
responses_ids, responses_ids,
) )
meta_info['turns_stats'] = turns_stats.tolist()
meta_info['active_mask'] = active_mask.tolist()
meta_info['valid_action_stats'] = valid_action_stats.tolist()
meta_info['valid_search_stats'] = valid_search_stats.tolist()
print("ACTIVE_TRAJ_NUM:", active_num_list) print("ACTIVE_TRAJ_NUM:", active_num_list)
return self._compose_final_output(original_left_side, original_right_side, meta_info) return self._compose_final_output(original_left_side, original_right_side, meta_info)