From 24ffaac722a157151b3727df21db190b4924d3e1 Mon Sep 17 00:00:00 2001 From: mckay Date: Tue, 19 Nov 2024 02:05:28 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20config=20=E8=B7=AF=E5=BE=84=E4=BD=BF?= =?UTF-8?q?=E7=94=A8=E7=BB=9D=E5=AF=B9=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- brep2sdf/config/default_config.py | 14 ++++++++++++-- brep2sdf/train.py | 1 - 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index a27e869..55f47bd 100644 --- a/brep2sdf/config/default_config.py +++ b/brep2sdf/config/default_config.py @@ -24,7 +24,17 @@ class DataConfig: brep_dir: str = '/home/wch/brep2sdf/test_data/pkl' sdf_dir: str = '/home/wch/brep2sdf/test_data/sdf' valid_data_dir: str = '/home/wch/brep2sdf/test_data/result/pkl' - save_dir: str = 'checkpoints' + + + # 保存路径 + save_dir: str = '/home/wch/brep2sdf/checkpoints' # 模型保存基础目录 + model_save_dir: str = '/home/wch/brep2sdf/checkpoints/models' # 模型文件保存目录 + result_save_dir: str = '/home/wch/brep2sdf/checkpoints/results' # 结果保存目录 + + # 文件命名 + model_name: str = 'brep2sdf' # 模型名称，用于文件命名 + checkpoint_format: str = '{model_name}_epoch_{epoch:03d}.pth' # 检查点文件名格式 + best_model_name: str = '{model_name}_best.pth' # 最佳模型文件名格式 @dataclass class TrainConfig: @@ -59,7 +69,7 @@ class LogConfig: log_interval: int = 10 # 本地日志 - log_dir: str = 'logs' # 日志保存目录 + log_dir: str = '/home/wch/brep2sdf/logs' # 日志保存目录 log_level: str = 'INFO' # 日志级别 console_level: str = 'INFO' # 控制台日志级别 file_level: str = 'DEBUG' # 文件日志级别 diff --git a/brep2sdf/train.py b/brep2sdf/train.py index da8484e..09854c7 100644 --- a/brep2sdf/train.py +++ b/brep2sdf/train.py @@ -15,7 +15,6 @@ def main(): # 创建所有保存目录 os.makedirs(config.data.model_save_dir, exist_ok=True) - os.makedirs(config.data.log_save_dir, exist_ok=True) os.makedirs(config.data.result_save_dir, exist_ok=True) # 初始化wandb