@ -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