Browse Source

config: change batch and epoch

main
mckay 3 months ago
parent
commit
3df0cbd19b
  1. 4
      brep2sdf/config/default_config.py

4
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

Loading…
Cancel
Save