print the config first

This commit is contained in:
2026-03-03 14:40:03 +08:00
parent 7294e07df7
commit 62a00b4a5b

View File

@@ -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()