diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index f800908..0a6cad0 100644 --- a/brep2sdf/config/default_config.py +++ b/brep2sdf/config/default_config.py @@ -55,9 +55,9 @@ class DataConfig: class TrainConfig: """训练相关配置""" # 基本训练参数 - batch_size: int = 1 + batch_size: int = 8 num_workers: int = 4 - num_epochs: int = 10 + num_epochs: int = 100 learning_rate: float = 1e-4 min_lr: float = 1e-6 weight_decay: float = 0.01