From 3df0cbd19b94019dc414a95bd4218a1fa75efa60 Mon Sep 17 00:00:00 2001
From: mckay <wchpub@163.com>
Date: Thu, 5 Dec 2024 01:17:13 +0800
Subject: [PATCH] config: change batch and epoch

---
 brep2sdf/config/default_config.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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