wx44wx/icassp2027-experiment
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow explicitly audited Ego4D bbox tails

Browse Source
This commit is contained in:
Xin Wang
2026-08-28 12:17:02 +08:00
parent 4011ba9b2a
commit 4abbde4f52
4 changed files with 180 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_ego4d_local_multigpu.py
+10
View File
@@ -64,6 +64,16 @@ class MultiGPUExtractionTests(unittest.TestCase):
command = plans[0]["command"]
self.assertEqual(command[command.index("--num-shards") + 1], "1")
def test_explicit_bbox_tail_limit_is_forwarded_to_all_workers(self):
option = "--ego4d-max-terminal-overrun-frames"
args, extraction, forwarded = launch.parse_args(self.argv + [option, "13"])
self.assertEqual(extraction.ego4d_max_terminal_overrun_frames, 13)
for plan in launch.plan_workers(args, extraction, forwarded):
command = plan["command"]
self.assertEqual(command[command.index(option) + 1], "13")
with contextlib.redirect_stderr(io.StringIO()), self.assertRaises(SystemExit):
launch.parse_args(self.argv + [option, "-1"])
def test_rejects_unsafe_or_ambiguous_launcher_options(self):
for extra in (["--gpus", "0", "0"], ["--gpus", "-1"], ["--cpu-threads", "0"],
["--preview-only"], ["--max-steps", "20"], ["--overwrite"],