|
@ -55,9 +55,9 @@ class DataConfig: |
|
|
class TrainConfig: |
|
|
class TrainConfig: |
|
|
"""训练相关配置""" |
|
|
"""训练相关配置""" |
|
|
# 基本训练参数 |
|
|
# 基本训练参数 |
|
|
batch_size: int = 1 |
|
|
batch_size: int = 8 |
|
|
num_workers: int = 4 |
|
|
num_workers: int = 4 |
|
|
num_epochs: int = 10 |
|
|
num_epochs: int = 100 |
|
|
learning_rate: float = 1e-4 |
|
|
learning_rate: float = 1e-4 |
|
|
min_lr: float = 1e-6 |
|
|
min_lr: float = 1e-6 |
|
|
weight_decay: float = 0.01 |
|
|
weight_decay: float = 0.01 |
|
|