diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index 55f47bd..12c0a18 100644 --- a/brep2sdf/config/default_config.py +++ b/brep2sdf/config/default_config.py @@ -40,7 +40,7 @@ class DataConfig: class TrainConfig: """训练相关配置""" # 基本训练参数 - batch_size: int = 32 + batch_size: int = 1 num_workers: int = 4 num_epochs: int = 100 learning_rate: float = 1e-4