huangfu/MAGAIL4AutoDrive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增修改

Browse Source
This commit is contained in:
huangfu
2026-02-04 21:07:09 +08:00
parent 95cc78d940
commit ceb6648a31
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/visualize.py
View File

@@ -55,7 +55,7 @@ def _run_replay(args):
print(f"Error resetting scenario {i}: {e}")
continue
print(f"Scenario loaded. Controlled agents: {len(env.controlled_agents)}")
print(f"Scenario loaded. Controlled agents (current): {len(env.controlled_agents)}, total in scenario: {env.num_controlled_in_scenario}")
for step in range(args.horizon):
obs, rewards, dones, infos = env.step(None)
@@ -176,7 +176,7 @@ def _run_policy(args):
pass
continue
print(f"Scenario loaded. Controlled agents: {len(obs_dict)}")
print(f"Scenario loaded. Controlled agents (current): {len(obs_dict)}, total in scenario: {env.num_controlled_in_scenario}")
if len(obs_dict) == 0:
print(f"Scenario {i} has no controlled agents (all filtered out). Skipping.")
continue