From 62a00b4a5bb17b4e134fa46562c33cfdf9c9495a Mon Sep 17 00:00:00 2001 From: goulustis Date: Tue, 3 Mar 2026 14:40:03 +0800 Subject: [PATCH] print the config first --- scripts/py_scripts/misc_tasks.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/scripts/py_scripts/misc_tasks.py b/scripts/py_scripts/misc_tasks.py index 595a85c..364a698 100644 --- a/scripts/py_scripts/misc_tasks.py +++ b/scripts/py_scripts/misc_tasks.py @@ -7,6 +7,7 @@ import os from tqdm import tqdm import yaml import tyro +from loguru import logger def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"): @@ -17,9 +18,11 @@ def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"): img = graph.show_graph(ret_img=True) img.save(osp.join(save_dir, f"arch_{conf.__class__.__name__}.png")) -def make_save_conf(pipeline_config:PipelineConfig, save_path:str): + +def make_save_conf(pipeline:PipelineConfig, save_path:str): os.makedirs(osp.dirname(save_path), exist_ok=True) - pipeline_config.save_config(save_path) + logger.info(pipeline) + pipeline.save_config(save_path) if __name__ == "__main__": # gen_arch_imgs()