From f6c312d140c718523e2ab4eabbc1b5bf1735a2c8 Mon Sep 17 00:00:00 2001 From: Alexander Raistrick Date: Tue, 8 Oct 2024 14:03:00 -0400 Subject: [PATCH] Move gin initialization inside slurm-submitted function, so slurm jobs recieve gin configs --- .../generate_individual_assets.py | 38 +++++++++---------- 1 file changed, 19 insertions(+), 19 deletions(-) diff --git a/infinigen_examples/generate_individual_assets.py b/infinigen_examples/generate_individual_assets.py index 6762b5b2..a4c85796 100644 --- a/infinigen_examples/generate_individual_assets.py +++ b/infinigen_examples/generate_individual_assets.py @@ -236,6 +236,25 @@ def build_and_save_asset(payload: dict): output_folder.mkdir(exist_ok=True) + init.apply_gin_configs( + ["infinigen_examples/configs_indoor", "infinigen_examples/configs_nature"], + configs=args.configs, + overrides=args.overrides, + skip_unknown=True, + ) + + if args.debug is not None: + for name in logging.root.manager.loggerDict: + if not name.startswith("infinigen"): + continue + if len(args.debug) == 0 or any(name.endswith(x) for x in args.debug): + logging.getLogger(name).setLevel(logging.DEBUG) + + init.configure_blender() + + if args.gpu: + init.configure_render_cycles() + logger.info(f"Building scene for {factory_name} {idx}") if args.seed > 0: @@ -454,25 +473,6 @@ def mapfunc( def main(args): bpy.context.window.workspace = bpy.data.workspaces["Geometry Nodes"] - init.apply_gin_configs( - ["infinigen_examples/configs_indoor", "infinigen_examples/configs_nature"], - configs=args.configs, - overrides=args.overrides, - skip_unknown=True, - ) - - if args.debug is not None: - for name in logging.root.manager.loggerDict: - if not name.startswith("infinigen"): - continue - if len(args.debug) == 0 or any(name.endswith(x) for x in args.debug): - logging.getLogger(name).setLevel(logging.DEBUG) - - init.configure_blender() - - if args.gpu: - init.configure_render_cycles() - if args.output_folder is None: outputs = Path("outputs") assert outputs.exists(), outputs