From fe9a5a68a730f1a6192fa29412ec63c5b9f12198 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