fix: config 路径使用绝对路径

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'                # 文件日志级别

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