config: 修改

brep2sdf/config/default_config.py

@@ -6,7 +6,7 @@ class ModelConfig:
     """模型相关配置"""
     brep_feature_dim: int = 16
     use_cf: bool = True
-    embed_dim: int = 384 # 3 的 倍数
+    embed_dim: int = 768 # 3 的 倍数
     latent_dim: int = 16
     
     # 点云采样配置
@@ -29,7 +29,7 @@ class DataConfig:
     """数据相关配置"""
     max_face: int = 32
     max_edge: int = 128
-    num_query_points: int = 4096  # 限制查询点数量，sdf 采样点数 本来是 128*128*128 ，在data load时随机采样
+    num_query_points: int = 32*32*32  # 限制查询点数量，sdf 采样点数 本来是 128*128*128 ，在data load时随机采样
     bbox_scaled: float = 1.0
 
     # pre_process
@@ -58,8 +58,8 @@ class TrainConfig:
     batch_size: int = 8
     num_workers: int = 4
     num_epochs: int = 100
-    learning_rate: float = 1e-4
-    min_lr: float = 1e-6
+    learning_rate: float = 1
+    min_lr: float = 1e-1
     weight_decay: float = 0.01
     
     # 梯度和损失相关