TODO: 太大加载不了

4 months ago · 2269a77f6a
1 changed files with 1 additions and 1 deletions
--- 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