print the config first
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user