From 8fe07b2b3ae8938f7f3645a6dadd2537bc2a13fc Mon Sep 17 00:00:00 2001 From: mckay Date: Mon, 31 Mar 2025 15:25:22 +0800 Subject: [PATCH] =?UTF-8?q?gitignore=20=E8=BF=98=E6=9C=89config=20?= =?UTF-8?q?=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .gitignore | 9 ++++++++- brep2sdf/config/default_config.py | 23 +++-------------------- 2 files changed, 11 insertions(+), 21 deletions(-) diff --git a/.gitignore b/.gitignore index 4eec394..53c5944 100644 --- a/.gitignore +++ b/.gitignore @@ -171,4 +171,11 @@ test_data/ logs/ wandb/ *.pth -checkpoints/ \ No newline at end of file +checkpoints/ + +data/gt_mesh +data/gt_point +data/step +data/input_data +data/output_data +data/name_list.txt \ No newline at end of file diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index 372f2eb..96e71bf 100644 --- a/brep2sdf/config/default_config.py +++ b/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