From 2269a77f6a1bba73386671fa5940cced46920fcc Mon Sep 17 00:00:00 2001 From: mckay Date: Tue, 19 Nov 2024 03:51:58 +0800 Subject: [PATCH] =?UTF-8?q?TODO:=20=E5=A4=AA=E5=A4=A7=E5=8A=A0=E8=BD=BD?= =?UTF-8?q?=E4=B8=8D=E4=BA=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- brep2sdf/config/default_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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