Skip to content

Commit

Permalink
Fix multiple cameras not running in paralell
Browse files Browse the repository at this point in the history
  • Loading branch information
araistrick authored and pvl-bot committed Oct 28, 2024
1 parent f4710c4 commit 4e7b9cb
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion infinigen/datagen/monitor_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand All @@ -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
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 4e7b9cb

Please sign in to comment.