From 22b1a54c2fad4082d48862516aa8cca30d674763 Mon Sep 17 00:00:00 2001 From: mckay Date: Fri, 13 Dec 2024 23:06:40 +0800 Subject: [PATCH] =?UTF-8?q?config:=20=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- brep2sdf/config/default_config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index 0a6cad0..2c19bde 100644 --- a/brep2sdf/config/default_config.py +++ b/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 # 梯度和损失相关