gitignore 还有config 修改

.gitignore

@@ -171,4 +171,11 @@ test_data/
 logs/
 wandb/
 *.pth
-checkpoints/
+checkpoints/
+
+data/gt_mesh
+data/gt_point
+data/step
+data/input_data
+data/output_data
+data/name_list.txt

brep2sdf/config/default_config.py

@@ -35,26 +35,10 @@ class DataConfig:
     # pre_process
     origin_brep_dir: str = '/mnt/mynewdisk/dataset/furniture/step/furniture_dataset_step/'
     
-    # 数据路径
-    pkl_path = "/home/wch/brep2sdf/test_data/pkl/train/bathtub_0004.pkl"
-    sdf_path = "/home/wch/brep2sdf/test_data/sdf/train/bathtub_0004.npz"
-
     # 基础路径配置
     base_data_dir: str = '/home/wch/brep2sdf/test_data'  # 基础数据目录
 
     
-    @property
-    def brep_dir(self) -> str:
-        return os.path.join(self.base_data_dir, 'pkl')
-    
-    @property
-    def sdf_dir(self) -> str:
-        return os.path.join(self.base_data_dir, 'sdf')
-    
-    @property
-    def valid_data_dir(self) -> str:
-        return os.path.join(self.base_data_dir, 'result/pkl')
-    
     
 
 @dataclass
@@ -84,10 +68,9 @@ class TrainConfig:
     # 保存路径
     checkpoint_dir: str = '/home/wch/brep2sdf/checkpoints'   # 结果保存目录
     
-    # 文件命名
+    # 文件命名                   # 模型名称，用于文件命名
-    model_name: str = 'brep2sdf'                     # 模型名称，用于文件命名
+    
-    checkpoint_format: str = '{model_name}_epoch_{epoch:03d}.pth'  # 检查点文件名格式
+    best_model_name: str = '{model_name}/best.pth'   # 最佳模型文件名格式
-    best_model_name: str = '{model_name}_best.pth'   # 最佳模型文件名格式
 
 
 @dataclass