From 4e7b9cbd20d3254f7f7f9214ac87b37bd69d3933 Mon Sep 17 00:00:00 2001 From: Alexander Raistrick Date: Sun, 20 Oct 2024 12:57:41 -0400 Subject: [PATCH] Fix multiple cameras not running in paralell --- infinigen/datagen/monitor_tasks.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/infinigen/datagen/monitor_tasks.py b/infinigen/datagen/monitor_tasks.py index 8eabfa40..cc707790 100644 --- a/infinigen/datagen/monitor_tasks.py +++ b/infinigen/datagen/monitor_tasks.py @@ -221,6 +221,8 @@ def iterate_scene_tasks( running_views = 0 for cam_rig, view_frame in itertools.product(cam_rigs, view_frames): + logger.debug(f"Checking {seed=} {cam_rig=} {view_frame=}") + view_frame_range = [ view_frame, min(frame_range[1], view_frame + view_block_size - 1), @@ -238,8 +240,11 @@ def iterate_scene_tasks( configs=global_configs, output_indices=view_idxs, ) + + state = JobState.Succeeded for state, *rest in view_tasks_iter: yield state, *rest + if state not in CONCLUDED_JOBSTATES: if viewdep_paralell: running_views += 1 @@ -285,7 +290,7 @@ def iterate_scene_tasks( input_indices=view_idxs if len(view_dependent_tasks) else None, output_indices={**camdep_indices, **extra_indices}, ) - + state = JobState.Succeeded for state, *rest in camera_dep_iter: yield state, *rest if state not in CONCLUDED_JOBSTATES: